一分钟解决python对dataframe的某列进行一定范围值的挑选

六个橙子_lc

已于 2024-04-16 15:44:25 修改

阅读量864

点赞数 3

CC 4.0 BY-SA版权

文章标签： python numpy pandas

于 2024-04-11 15:02:23 首次发布

本文链接：https://blog.youkuaiyun.com/lc_lcrystal/article/details/137639194

本文介绍了如何在Python中使用pandas的loc方法和numpy的where函数，针对DataFrame的特定列（如cloud列）筛选出值在20到80范围内的行，展示了数据处理中的条件筛选过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、解释：在Python 中可以使用 pandas.DataFrame.loc和 numpy.where来实现
2、示例：
假设我们有一个名为data_cloud的DataFrame，其中有一列名为cloud，我们想要筛选出cloud列中值为在20~80范围的所有行。可以使用以下代码实现：

import pandas as pd
# 创建一个示例DataFrame，cloud表示云量
data_cloud = {'A': [1, 2, 3, 4, 5],
        'B': ['district1', 'district2', 'district3', 'district4', 'district5'],
        'cloud': [20, 10, 40, 60,90]}
data_cloud_wea = pd.DataFrame(data_cloud)

# 使用numpy.where进行条件筛选，判断是否是多云天气
data_cloud_wea.loc[:, 'weather'] = np.where((data_cloud_wea.loc[:, "cloud"] > 25) & (data_cloud_wea.loc[:, "cloud"] < 70), '多云', '非多云')
print(data_cloud_wea)