Question

我有一个字典，它使用字符串（编辑）作为键，并将列表列表存储为值。

dict = {key1: [[data1],[data2],[data3]], key2: [[data4],[data5]],...etc}

编辑：数据变量是包含转换后的pandas DataFrame

中不同数据类型的行

实施例

df = pd.DataFrame()
df['City'] = ['New York','Austin','New Orleans','New Orleans']
df['State'] = ['NY','TX','LA','LA']
df['Latitude'] = [29.12,23.53,34.53,34.53]
df['Time'] = [1.46420e+09,1.47340e+09,1.487820e+09,1.497820e+09]

City         State    Latitude   Time
New York     NY       29.12      1.46420e+09
Austin       TX       23.53      1.47340e+09
New Orleans  LA       34.53      1.487820e+09
New Orleans  LA       34.53      1.497820e+09

dict = {}
cities = df['City'].unique()
for c in cities:
    temp = df[df['City'] == c]
    dict[c] = temp.as_matrix().tolist()

#which outputs this for a given key
dict['New Orleans'] = [['New Orleans' 'LA' 34.53  1.487820e+09],
    ['New Orleans' 'LA' 34.53  1.497820e+09]]

我使用以下内容将其存储为csv：

filename = 'storage.csv'
with open(filename,'w') as f:
    w = csv.writer(f)
    for key in dict.keys():
        w.writerow((key,dict[key]))

然后我使用以下内容将文件读回字典：

reader = csv.reader(open(filename, 'r'))
dict = {}
for key,val in reader:
    dict[key] = val

除了它现在是一个字符串之外，val看起来很完美。例如，key1如下所示：

dict[key1] = "[[data1],[data2],[data3]]"

如何读取列表中的值，或从val的读入版本中删除引号？

Answer 1

修改：由于您使用的是pandas.DataFrame，请不要使用csv模块或json模块。相反，使用pandas.io进行阅读和写作。

原始答案：

简答：使用json。

CSV可以保存字符串表。除此之外还需要手动将字符串转换回Python对象。

如果您的数据只是列表，那么词典和基本文字（如字符串和数字json）将是此工作的正确工具。

假设：

example = {'x': [1, 2], 'y': [3, 4]}

保存到文件：

with open('f.txt','w') as f:
    json.dump(example, f)

从文件加载：

with open('f.txt') as f:
    reloaded_example = json.load(f)

Answer 2

您的代码必须如下：

import csv
import ast
#dict = {1: [[1],[2],[3]], 2: [[4],[5]]}
reader = csv.reader(open("storage.csv", 'r'))
dict = {}
for key,val in reader:
    dict[int(key)] = ast.literal_eval(val)
print dict

python - 列表正在从csv作为字符串读入

2 个答案: