python - 将一列的每个元素乘以同一数据框中不同列的每个元素

Question

我需要将一列的每个元素乘以来自同一数据帧的不同列的每个元素。我的原始数据集看起来像这样:

   origin    sum    sum2
    a.        2      1
    b.        4      2
    c.        6      3

我期望的结果类似于:

   origin    dest   result (sum * sum2)
    a.        a.      2
    a.        b.      4
    a.        c.      6
    b.        a.      4
    b.        b.      8
    b.        c.      12
    c.        a.      6
    c.        b.      12
    c.        c.      18

我正在写的脚本如下，但是我得不到需要的结果:

x = 0
numerator = []

for index1, row1 in df.iterrows():
    constant = row1
    numerator.append([])

    for index2, row2 in df.iterrows():
        result = row2*constant

        numerator[x].append(result)

        x = x + 1

Answer 1

您可以使用:

• numpy.outer对于多个

• numpy.ravel用于展平

• MultiIndex.from_product从列 origin

获取新索引

• DataFrame构造函数

• reset_index对于 MultiIndex 中的列:

---

mux = pd.MultiIndex.from_product([df.origin, df.origin], names=['origin','dest'])
data = np.outer(df['sum'], df['sum2']).ravel()
df = pd.DataFrame(data, index=mux, columns=['result']).reset_index()
print (df)
  origin dest  result
0     a.   a.       2
1     a.   b.       4
2     a.   c.       6
3     b.   a.       4
4     b.   b.       8
5     b.   c.      12
6     c.   a.       6
7     c.   b.      12
8     c.   c.      18