如何过滤列并创建新DataFrame?请编写过滤生成df2的代码
1. 过滤掉某一列并创建新DataFrame的方法
你可以通过以下两种简单方式实现:
- 使用
drop()方法移除指定列,返回新DataFrame(不修改原数据):# 移除名为"unwanted_col"的列 new_df = original_df.drop("unwanted_col", axis=1) - 直接选择需要保留的列,间接排除目标列:
# 仅保留"col_a"、"col_b"、"col_c"三列 new_df = original_df[["col_a", "col_b", "col_c"]]
2. 提取purpose列值为"house"的记录创建df2
利用布尔索引筛选符合条件的行即可:
# 筛选df1中purpose等于"house"的所有行,赋值给df2 df2 = df1[df1["purpose"] == "house"]
如果需要重置df2的索引(避免继承原df1的索引),可以加上reset_index():
df2 = df1[df1["purpose"] == "house"].reset_index(drop=True)
内容的提问来源于stack exchange,提问作者Dayssy Moreno
相关产品推荐
相关产品推荐

