python - 将数据框中的每一列与其右侧的列相乘 Python

Question

我有一个大型数据框，其中包含大约 400.000 个观察值和 6.500 列。我正在寻找一种快速方法，将每一列依次与其右侧的列相乘。

示例数据框可能如下所示:

| V1  | V2  | V3  |  
----------------------
|  1  |  2  |  1  |
|  0  |  4  |  1  |
|  1  |  3  |  3  |

最后我想要这样的东西:

| V1 | V2 | V3 | V1_V2 | V1_V3 | V2_V3 |
-----------------------------------------
|  1 |  2 |  1 |    2  |   1   |   2   |
|  0 |  4 |  1 |    0  |   0   |   4   |
|  1 |  3 |  3 |    3  |   0   |   9   |

我尝试了 itertools.combinations 但它太慢了。我是 Python 的初学者，所以也许有一个我不知道的简单解决方案。

感谢您的帮助!

Answer 1

pandas 向量化运算(如乘法)本身非常高效。您可以使用类似以下的方法来利用这一点:

# Extract column names
cols =  df.columns.tolist() 

# Generate all adjacent pairs, including the circular one
cols_to_create = [(cols[i], cols[i+1]) for i in range(len(cols)-1)] \
                 + [(cols[len(cols)-1], cols[0])] 

# Perform multiplication on all pairs
for x, y in cols_to_create:
    df[x+'_'+y] = df[x]*df[y]