我有一个数据框，其中一列在每个索引处都有一个列表。我想将这些列表连接成一个列表。我正在使用
ids = df.loc[0:index, 'User IDs'].values.tolist()
但是，这会导致
['[1,2,3,4......]'] 这是一个字符串。不知何故，我的列表列中的每个值都是str 类型。我曾尝试使用list()、literal_eval() 进行转换，但它不起作用。 list() 将列表中的每个元素转换为字符串，例如从[12,13,14...] 到['['1'',','2',','1',',','3'......]']。
如何将 pandas 列与列表值连接到一个列表中？请帮忙，我已经敲了几个小时。
【问题讨论】：
考虑数据框df
df = pd.DataFrame(dict(col1=[[1, 2, 3]] * 2))
print(df)
col1
0 [1, 2, 3]
1 [1, 2, 3]
pandas最简单的答案df.col1.sum()
[1, 2, 3, 1, 2, 3]
numpy.concatenatenp.concatenate(df.col1)
array([1, 2, 3, 1, 2, 3])
chainfrom itertools import chain
list(chain(*df.col1))
[1, 2, 3, 1, 2, 3]
对 cmets 的回应：
我认为您的列是字符串
from ast import literal_eval
df.col1 = df.col1.apply(literal_eval)
如果您的列是看起来像列表的字符串值
df = pd.DataFrame(dict(col1=['[1, 2, 3]'] * 2))
print(df) # will look the same
col1
0 [1, 2, 3]
1 [1, 2, 3]
但是pd.Series.sum 的工作方式不同。
df.col1.sum()
'[1, 2, 3][1, 2, 3]'
我们需要评估字符串，就好像它们是文字一样，然后sum
df.col1.apply(literal_eval).sum()
[1, 2, 3, 1, 2, 3]
【讨论】：
df.col1.iloc[:3].sum() 仅合并前 3 个。
ids = pd.DataFrame(GCM.loc[0:2, 'User IDs']) ids = uninstall_ids['User IDs'].sum() 但问题是它的类型是 str ，这会在程序中进一步产生问题。如果我使用 list() 或 tolist() 那么它将 [12,13,14,15] 转换为 ['['1',',','2',',','1',', ','3'...]']。我需要这是一个列表，因为我正在使用集合中的 counter() 来与程序稍后的两个列表进行比较。我也试过df.col1.iloc[:3].sum() 但同样的问题。
如果您想展平列表，这是pythonic 的方法：
将熊猫导入为 pd
df = pd.DataFrame({'A': [[1,2,3], [4,5,6]]})
a = df['A'].tolist()
a = [i for j in a for i in j]
print a
【讨论】：