从嵌套列表创建新列表并在中转换str

2024-04-25 07:31:44 发布

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

很抱歉问了一个以前被问过一百次的问题,我是Python新手,我找到的解决方案似乎都不能解决我的问题。你知道吗

我有一个csv文件名为diabetes的嵌套列表,我读入该文件并用逗号分隔如下元素

for line in open("diabetes.csv"):
    lst=line.strip().split(",")
    print(lst)

打印出以下内容

['10', '101', '86', '37', '0', '45.6', '1.136', '38', '1']
['2', '108', '62', '32', '56', '25.2', '0.128', '21', '0']
['3', '122', '78', '0', '0', '23', '0.254', '40', '0']

现在我的问题是

  1. 我需要制作一个单独的列表,只包含每个列表的第三个元素(lst[2])
  2. 我需要把它转换成浮点数而不是字符串。你知道吗

我使用的是python3.6,我在这里把头发拔出来。你知道吗


Tags: 文件csvin元素列表for文件名line
3条回答

假设您有一个字符串列表:

LoL=[
   ['10', '101', '86', '37', '0', '45.6', '1.136', '38', '1'],
   ['2', '108', '62', '32', '56', '25.2', '0.128', '21', '0'],
   ['3', '122', '78', '0', '0', '23', '0.254', '40', '0'],
]

您可以获得每个子列表的nth元素,如下所示:

>>> [float(sl[2]) for sl in LoL]
[86.0, 62.0, 78.0]

如果您有csv文件,请使用csv module执行完全相同的操作:

(在命令提示下):

$ cat file.csv
10,101,86,37,0,45.6,1.136,38,1
2,108,62,32,56,25.2,0.128,21,0
3,122,78,0,0,23,0.254,40,0

Python:

import csv
with open('file.csv') as f:
  items=[float(row[2]) for row in csv.reader(f)]

>>> items
[86.0, 62.0, 78.0]

所以——底线是:

  1. 请使用csvpandas而不是.split(','),以便您可以正确处理引用的csv和其他特殊性
  2. 使用with上下文管理器,使文件在块的末尾自动关闭
  3. csv文件非常类似于列表列表,通常可以用相同的方式处理。你知道吗

非常简单和天真的一句话:

result = [float(line.strip().split(",")[2]) for line in open("diabetes.csv")]

您可以使用pandas模块,这在数据科学中是相当标准的:

import pandas as pd

df = pd.read_csv("diabetes.csv", header=None, index_col=None)
df.iloc[:, 2] = pd.to_numeric(df.iloc[:, 2], downcast='float')
list = df.iloc[:, 2]

编辑 注意,这里的输出类型是Series

相关问题 更多 >

    热门问题