假设你有一个 pandas DataFrame,它包含了一些数据,你想要根据某些特定条件筛选出其中的一些列,可以使用以下方法:

import pandas as pd

# 创建一个 DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David', 'Eric'],
        'Age': [25, 32, 18, 47, 22],
        'Gender': ['F', 'M', 'M', 'M', 'M'],
        'Salary': [50000, 60000, 40000, 80000, 35000]}
df = pd.DataFrame(data)

# 筛选出年龄大于等于30岁的员工的姓名和工资
result = df.loc[df['Age'] >= 30, ['Name', 'Salary']]
print(result)

输出结果:

     Name  Salary
1     Bob   60000
3   David   80000

在这个例子中,我们首先创建了一个 DataFrame,然后使用 loc 方法根据条件筛选出了年龄大于等于 30 岁的员工,并且只选择了这些员工的姓名和工资两列。最后将筛选结果打印出来。


原文地址: http://www.cveoy.top/t/topic/mYY 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录