在单元格python中计算数组

Question

我的数据框中有数组需要计算。

这是我使用的代码：

indi = data_1.query("'2016-11-22' <= login_date <= '2016-12-22'").groupby(['employer_key','account_id','login_date']).count().reset_index()
indi_1 = indi.groupby(['employer_key']).account_id.unique().reset_index()
indi_1

给了我这个：

    employer_key        account_id
0   boeing              [17008601, 17008645, 17008698, 17008952, 17009...]
1   dell_inc            [10892711, 10892747, 10894032, 10894676, 10894...]
2   google              [9215462, 9216605, 9217052, 9218693, 9222937, ...]
3   sprint_corporation  [9858036, 9858809, 9859191, 9859350, 9859498, ...]
4   walmart             [2515412, 2517367, 2519765, 2520049, 2526763, ...]

我想计算数组中的数字，看起来像这样：

  employer_key         account_id
0   boeing             5000
1   dell_inc           289
2   google             789
3   sprint_corporation 154670
4   walmart            4689

我该怎么做？我正在使用熊猫。我对python也很陌生，越简单越好。

Answer 1

如果 account_id 列包含列表，您可以使用str.len()计算每个单元格中的元素数量：

df['account_id_count'] = df.account_id.str.len()
df