python - 读取 CSV 的单列并存储在数组中

Question

从 csv 中读取但只有一个特定列(如 title)的最佳方法是什么？

ID | date|  title |
-------------------
  1|  2013|   abc |
  2|  2012|   cde |

然后该列应该存储在这样的数组中:

data = ["abc", "cde"]

这是我目前所拥有的，pandas :

data = pd.read_csv("data.csv", index_col=2)

我调查了 this thread .我仍然收到 IndexError: list index out of range。

编辑:

不是表格，是逗号分隔的:

ID,date,title
1,2013,abc
2,2012,cde

Answer 1

一个选项是读取整个 csv，然后选择一列:

data = pd.read_csv("data.csv")

data['title']  # as a Series
data['title'].values  # as a numpy array

正如@dawg 所建议的那样，如果您还使用 squeeze 参数来避免某些 hackery 压扁值数组，则可以使用 usecols 参数...

In [11]: titles = pd.read_csv("data.csv", sep=',', usecols=['title'], squeeze=True)

In [12]: titles  # Series
Out[12]: 
0    abc
1    cde
Name: title, dtype: object

In [13]: titles.values  # numpy array
Out[13]: array(['abc', 'cde'], dtype=object)