python - 如何将 Pandas DataFrame 的每一列与其他列相乘？

Question

给定一个 pandas 数据框，我想将每一列与其他列一一相乘，并将每个新列作为新列返回到该数据框。例如

A B C
1 2 3
2 4 4
1 2 5

然后

A B C A*B   A*C     B*C
1 2 2  2     3       6
2 4 8  8     8       16
1 2 2  2     5       10

Answer 1

以下是一种暴力方法，但它应该可以完成工作。 permutations() 生成所有列排列。内部的 sorted() 与 set() 一起将 ('A','B') 与 ('B', 'A') 等

import pandas as pd
import itertools

df = pd.DataFrame([[1,2,1],[2,4,2],[3,4,5]],columns=['A','B','C'])

for c1,c2 in sorted(set([tuple(sorted(s)) for s in itertools.permutations(df.columns,2)])):
  df['{0}x{1}'.format(c1,c2)] = df[c1]*df[c2]

print df