DataFrame筛选多行和多列

发布时间 2023-07-25 12:25:58作者: Bonne_chance

Dataframe筛选多行

在实际数据筛选的时候,有时候需要选择多行,例如,有一个列表数据,需要在Dataframe里筛选,某列中在列表数据中的行。
在这种情况下可以使用isin语法。具体如下:

obj_df = df[df['obj_col'].isin(obj_list)] # 筛选的行
non_obj_df = df[~df['obj_col'].isin(obj_list)] # 剩余的行

Dataframe筛选多列

方式1:知道要选择的列名,直接选择
obj_df = df[['col1','col2']]

方式2:按照索引,筛选列
obj_df = df[df.columns[0: obj_index]] #本质上和方式1相近