python - 为熊猫中的所有列生成列矩阵

Question

我有一个由 6 列组成的数据框。生成执行以下操作的矩阵的最快方法是什么：

步骤 1) col1*col1a , col2*col2a, col3*col3a, col4*col4a

步骤 2) col_new = (col1*col1a)-col2*col2a)/(col1a-col2a)

使用 for 循环是其中一种选择 - 但有什么方法可以更快地解决这个问题。

import pandas as pd
df=pd.DataFrame()
df['col1']=[100,200,300,400,500]
df['col1a']=[6,71,8,90,10]
df['col2']=[600,700,800,1900,100]
df['col2a']=[6,17,8,9,10]
df['col3']=[100,220,300,440,500]
df['col3a']=[1,22,3,44,5]

df[1x2]=(df['col1']*df['col1a']-df['col2']*df['col2a'])/(df['col1a']-df['col2a'])

我需要有 1x3,1x4,1x5,2x3,2x4 等的列组合...

标签： pythonpandasdataframepython-3.6

Answer 1

这是我将如何处理它：

def new_col(df, col1, col2):
    """
    Add a new column, modifying the dataframe inplace.

    col1: int
        column counter in the first column name
    col2: int
        column counter in the second column name
    """
    nr = (
        df.loc[:, f"col{col1}"] * df.loc[:, f"col{col1}a"]
        - df.loc[:, f"col{col2}"] * df.loc[:, f"col{col2}a"]
    )
    dr = df.loc[:, f"col{col1}a"] - df.loc[:, f"col{col2}a"]

    df.loc[:, f"col{col1}X{col2}"] = nr / dr

我将使用所需的列组合调用此函数。例如。

new_col(df, 1, 2)

输出：

从循环发出调用。