在Python中从子列表获取元素

2024-05-19 09:33:54 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个.csv文件,其中包含以下4行元素:

dev1, NX-OS, 10.0.0.1, admin, admin
dev2, NX-OS, 10.0.0.2, admin, admin
dev3, NX-OS, 10.0.0.3, admin, admin
dev4, NX-OS, 10.0.0.4, admin, admin

我试图提取任何元素。为了做到这一点,我有以下几点:

import csv
dev_list=[]
infile=open('dev_in.csv','r')
csv_in=csv.reader(infile)

for dev in csv_in:
    dev_list.append(dev)

dev_list的结果是以下列表:

[
    ['dev1, NX-OS, 10.0.0.1, admin, admin'],
    ['dev2, NX-OS, 10.0.0.2, admin, admin'],
    ['dev3, NX-OS, 10.0.0.3, admin, admin'],
    ['dev4, NX-OS, 10.0.0.4, admin, admin']
]

为了提取第一个列表的第一个元素,我编码:

element=dev_list[0]
print(element[0])

但是,我得到了完整的第一个列表:

dev1, NX-OS, 10.0.0.1, admin, admin

也许我应该使用“split”函数将其拆分,但我认为这会不必要地扩展代码。我需要提取每个子列表的第一个元素。但是,我可能需要稍后提取其他元素

还有别的办法吗?还有没有更好的方法来编写这段代码,也许可以通过列表理解提前感谢


Tags: csvindev元素列表dev1adminos
1条回答
网友
1楼 · 发布于 2024-05-19 09:33:54

总结我的评论,下面是您需要做的事情(考虑到Python 3):

import csv
from pathlib import Path

dev_in = Path('dev_in.csv')

devs = list(csv.reader(dev_in.open(), skipinitialspace=True))

for dev in devs:
    # print dev names
    print(dev[0])

上述代码将给出以下行作为输出:

dev1
dev2
dev3
dev4

以下是概念证明:

Python 3.7.5 (default, Dec 15 2019, 17:54:26) 
[GCC 9.2.1 20190827 (Red Hat 9.2.1-1)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import csv
>>> from pathlib import Path
>>> dev_in = Path('dev_in.csv')
>>> 
>>> devs = list(csv.reader(dev_in.open(), skipinitialspace=True))
>>> devs
[['dev1', 'NX-OS', '10.0.0.1', 'admin', 'admin'], ['dev2', 'NX-OS', '10.0.0.2', 'admin', 'admin'], ['dev3', 'NX-OS', '10.0.0.3', 'admin', 'admin'], ['dev4', 'NX-OS', '10.0.0.4', 'admin', 'admin']]
>>> dev=devs[0]
>>> dev
['dev1', 'NX-OS', '10.0.0.1', 'admin', 'admin']
>>> dev[0]
'dev1'
>>>

我希望有帮助

相关问题 更多 >

    热门问题