[人工智能]pandas Dataframe 其中2列取小(大)值合成新列

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()

输出

(1000, 3)
? ? ? ? ? A? ? ? ? ? ? ? ?B? ? ? ? ? ? ? ?C? ? ? ? ? ? ? ?D? ? ? ? ? ? ?E
0 -0.879277 -1.027903 -0.602419 -1.027903 -0.602419
1 ?1.010298 ?1.604261 -0.097757 ?1.010298 ?1.010298
2 -0.228731 ?0.646538 ?0.020998 -0.228731 ?0.020998
3 ?1.404384 -1.681164 ?0.134697 -1.681164 ?1.404384
4 -0.368503 ?0.270716 ?1.419268 -0.368503 ?1.419268
5 ?2.238605 ?1.628516 -1.984287 ?1.628516 ?2.238605
6 ?0.652273 -0.166380 ?1.855101 -0.166380 ?1.855101
7 ?0.602067 ?0.604722 -0.640275 ?0.602067 ?0.602067
8 -2.775687 -2.649085 ?0.502292 -2.775687 ?0.502292
9 ?0.031345 ?0.419377 -0.445941 ?0.031345 ?0.031345