如何在python中基于条件从pandas数据帧中选择列

import pandas as pd

data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10],
        'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)

condition = df['A'] > 2

selected_columns = df[condition]

import pandas as pd

data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10],
        'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)

condition = df['A'] > 2
selected_columns = df[condition]

print(selected_columns)

   A   B   C
2  3   8  13
3  4   9  14
4  5  10  15