繁体   English   中英

如何使用 python 从 csv 文件中提取列表列表?

[英]How to extract list of lists from a csv-file with python?

我有一个 csv 文件,基本上类似于这种结构(1 列,2 行):

参与者,日期和时间,yadayada,...
{'Name': ...},...,"[[1, 1, 1, 0],[1, 0, 1, 1]]",

我想要做的是只检索被引号包围的列表列表。

import pandas as pd

with open('file.csv', 'w') as f:
    f.write(
        """participant,a,b,c,colname\n"""
        """{'Name': 'asdf'},3,5,6,"[[1, 1, 1, 0],[1, 0, 1, 1]]"\n"""
    )

df = pd.read_csv('file.csv')
colname = 'colname'

res = []
for sublist in df.loc[0, colname][2:-2].split('],['):
    res.append([int(item) for item in sublist.split(', ')])
print(l)

印刷

[[1, 1, 1, 0], [1, 0, 1, 1]]

备注:这里我假设数据中只有一行,并且您要访问的列存储在变量colname中。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM