python - pandas 提取数组到列

Question

数组(数组的长度对于系列中的所有元素都是常量)如何有效地提取到列中？

import pandas as pd
d = pd.DataFrame({'foo':\[1,2,3\], 'bar':\[\[1,1,1\], \[2,2,2\], \[3,3,3\]\]})
d][1]][1]

即将 [1,1,1] 的数组提取到 bar_0, bar_1, bar_3 列中？

有没有比手动遍历数组中的索引并调用 pandas.apply 更好的方法？

Answer 1

import pandas as pd

d = pd.DataFrame({"foo": [1, 2, 3], "bar": [[1, 1, 1], [2, 2, 2], [3, 3, 3]]})

d = pd.concat([d, d.pop("bar").apply(pd.Series).add_prefix("bar_")], axis=1)
print(d)

打印:

   foo  bar_0  bar_1  bar_2
0    1      1      1      1
1    2      2      2      2
2    3      3      3      3