python - 将可变大小的数组写入 Pandas 单元格

Question

我有一个大数据集，我想使用符合条件的多行进行卷积计算。我需要先为每一行计算一个向量，我认为将我的向量存储在数据框列中会更有效，这样我就可以在进行卷积时尝试避免 for 循环。问题是，向量是可变长度的，我不知道该怎么做。

这是我的数据摘要:

Date        State  Alloc P
2012-01-01  AK     3     0.5
2012-01-01  AL     4     0.3
…

每个状态都有不同的Alloc和P值。每个日期和状态都有一行，我的数据框超过 15,000 行。

对于每个条目，我想要一个如下所示的向量:

[P, np.zeros(Alloc), 1-P]

我不知道如何设置这样的新列。我试过这样的陈述:

df['Test'] = [df['P'], np.zeros(df['Alloc'), 1 – df['P']]

但它们不起作用。

有没有人有什么想法？

谢谢☺

Answer 1

尝试:

def get_vec(x):
    return [x.P] + np.zeros(x['Alloc']).tolist() + [1 - x.P]

df.apply(get_vec, axis=1)

0         [0.5, 0.0, 0.0, 0.0, 0.5]
1    [0.3, 0.0, 0.0, 0.0, 0.0, 0.7]
dtype: object

---

df['Test'] = df.apply(get_vec, axis=1)
df