过滤numpy数组中的值并获取DataFrame行的快速方法

import numpy as np
import pandas as pd

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
df = pd.DataFrame(arr, columns=['A', 'B', 'C'])

condition = arr[:, 0] > 3  # 这里以第一列大于3为例
filtered_rows = np.where(condition)[0]

filtered_df = df.iloc[filtered_rows]

import numpy as np
import pandas as pd

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
df = pd.DataFrame(arr, columns=['A', 'B', 'C'])

condition = arr[:, 0] > 3
filtered_rows = np.where(condition)[0]
filtered_df = df.iloc[filtered_rows]

print(filtered_df)