从Dataframe comumn创建数组

Question

我从Python开始。我想创建一个新的数组，它将包含从数据帧中的一个列存储的所有单词。

此列已包含带有单词的数组：

这里有一个例子：

df['Body_A'].values[0]

输出：

['p','open','terminal‘,'application','gt','utility','gt','terminal’,'type','p','pre','code','locate','insertfonthere‘,'br','code','pre','p','spit','every’,'file','p','p','warning‘,'may','alot','wade','p']

我试着把所有的东西都储存起来：

rows = []
[rows + row['Body_A'] for index, row in df.iterrows()]

我的想法是将数据帧中每一行的值存储在一个新的数组行中。

但当我要打印行时：

print(rows)

它还我一个空的我不知道我做错了什么可以帮助告诉我更多？

谢谢.

Answer 1

您可以使用嵌套列表理解将列表扁平化

df = pd.DataFrame({"a":[['Hello', 'I', 'am', 'fine'], ['How', 'are', 'you', '?']]})
flat_list = [item for sublist in df['a'] for item in sublist] 

df::
    a
0   [Hello, I, am, fine]
1   [How, are, you, ?]

输出：

['Hello', 'I', 'am', 'fine', 'How', 'are', 'you', '?']