分隔csv文件的不同部分？

Question

我是新手，我有一个 csv 文件，其中包含 reddit 的帐户名称、subreddit、时间和消息。

我是这样读的：

train_data = pd.read_csv("addres/train_data.csv", encoding="utf8")

如果我写： train_data.head()我明白了

你知道我可以用 [author, body] 创建一个数组的方法吗？

首先我尝试以这种方式创建两个数组（作者和消息）：

train=open("addres/train_data.csv")
train.readline()
author=[]
message=[]
for line in train:
    autore,categoria,ora, messaggio=line.split(",")
    author.append(autore)
    message.append(messaggio)

但是messages包含“，”所以它不能正常工作。

谢谢你，很抱歉这个愚蠢的问题。

Answer 1

df_tmp = train_data[['author', 'body']] # allows you to select subset by column name
content_array = [list(x) for x in df_tmp.values] # a list of lists ([ith_author, ith_body])

请记住，df_tmp 只是 train_data 的 view，您可能需要根据您打算如何使用它来创建 copy。如果您需要数据不可变，则可以使用tuples。