pandas Dataframe 其中2列取小(大)值合成新列

阅读 101

2022-03-31

import pandas as pd
import numpy as np


def DataframeMinTest():
    df = pd.DataFrame(np.random.randn(1000, 3), columns=list('ABC'))
    print(df.shape)
    # 方式一 取AB两列小值合成D列
    df['D'] = df.loc[:, ['A', 'B']].min(axis=1)
    # 方式二 取AC两列大值合成E列
    df['E'] = np.max(df.loc[:, ['A', 'C']].values, axis=1)
    print(df.head(10))


if __name__ == '__main__':
    DataframeMinTest()

输出

精彩评论(0)

0 0 举报