python - DataFrame Panda Python中的数组

Question

我有这个数据框。在 Column ArraysDate 中包含许多元素。我希望能够在 java 数组中编号并运行 for 循环。我还没有找到任何解决方案，请告诉我一些想法？例如 CustomerNumber = 4 ，则 ArraysDate 有 3 个元素，并且理解 i1、i2、i3、i4 以在 ArraysDate 中使用计算。谢谢

CustomerNumber           ArraysDate
1                         [ 1 13 ]
2                         [ 3 ]
3                         [ 0 ]
4                         [ 2 60 30 40]

标签： pythonpandas

Answer 1

您可以使用 lambda 对数组中每行的所有项目求和。

第 1 步：创建数据框

import pandas as pd
import numpy as np
d = {'ID': [[1,2,3],[1,2,43]]}
df = pd.DataFrame(data=d)

步骤 2：对数组中的项目求和

df['ID2']=df['ID'].apply(lambda x: sum(x))
df