如何使用python從csv文件中讀取數組列表

Question

我有一個帶“;”的csv文件 看起來像信息的分隔符，

Id;arrVal
0;[[84, 109, 423, 466]]
1;[[242, 192, 822, 594], [242, 640, 818, 911]]
2;[[123, 212, 401, 351]]

我正在讀取csv文件，並希望通過使用以下代碼來獲取特定ID的數組

df = pd.read_csv("path/to/csv/file.csv", sep=";")
df[df['Id']==1].iloc[0]['arrVal']

它給我的值是[[242, 192, 822, 594], [242, 640, 818, 911]]但是是字符串。 如何提取為2D數組？

Answer 1

該列也是有效的JSON語法，因此您可以使用：

import pandas as pd
import json

df = pd.read_csv('input.csv',sep=';')
L = json.loads(df[df['Id']==1].iloc[0]['arrVal'])

Answer 2

使用ast.literal_eval ：

x = ast.literal_eval(df[df['Id']==1].iloc[0]['arrVal'])