根据特定条件从python中的dataframe中随机选择行

import pandas as pd
import numpy as np

data = {'A': [1, 2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50],
        'C': ['a', 'b', 'c', 'd', 'e']}

df = pd.DataFrame(data)
print(df)

selected_rows = df.query('A > 2')
random_row = selected_rows.sample(n=1) # 设置n为所需的随机行数
print(random_row)

selected_rows = df[df['A'] > 2]
random_row = selected_rows.sample(n=1)  # 设置n为所需的随机行数
print(random_row)