從 Dataframe comumn 創建一個數組

Question

我從 Python 開始。 我想創建一個新數組，它將包含 dataframe 中一列中存儲的所有單詞。

此列已包含帶有單詞的數組：

這是一個例子：

df['Body_A'].values[0]

output：

['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' , '很多', '涉水', 'p']

我試圖存儲所有這樣做的東西：

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

我的想法是將 dataframe 中每一行的值存儲在一個新的數組行中。

但是當我想打印行時：

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, ?]

Output：

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