python - Python：如何像矩阵一样访问 CSV？

Question

我不想手动执行解析 CSV 的部分，我需要以这种方式访问​​单元格：

Cells (row, column) = X

或者

X = Cells (row, column)

有谁知道这是怎么做到的吗 ？

Answer 1

根据您需要的复杂程度，熊猫可能会很好。

import pandas as pd
location = r'C:\path\to\your\file.csv'
df = pd.read_csv(location, names=['Names','Births'])

df['Births'].plot()

Answer 2

numpy 很好，但对于这样一个简单的要求来说有点矫枉过正。

尝试：

import csv
sheet = list(csv.reader(open(source_csv_location)))
print sheet[0][0]

Answer 3

假设您有一个 CSV 文件并希望将其视为一个数组：

您可以genfromtxt从numpy模块中使用，这将创建一个 numpy 数组，其中包含与文件中一样多的行和列（X在下面的代码中）。假设数据都是数字，您可以使用它savetxt来将值存储在 csv 文件中：

import numpy as np
X = np.genfromtxt("yourcsvfile.dat",delimiter=",")
X[0,0] = 42   # do something with the array
np.savetxt('yourcsvfile.dat',X,delimiter=",")

编辑：

如果文件中有字符串，您可以这样做：

# read in
X = np.genfromtxt("yourcsvfile.dat",delimiter=",",dtype=None)
# write out
with open('yourcsvfile.dat','w') as f:
for el in X[()]:
    f.write(str(el)+' ') 

这里的答案中的一些其他技术：

numpy 将不同类型的数组保存到文本文件中

如何将不同类型的数据从文件导入 Python Numpy 数组？