Pandas中如何按Year列条件计算生成TrueEarn新列?
解决方法
你的代码语法不符合Python和Pandas的规则,下面给你两种简单可行的实现方式:
方法一:使用Pandas的loc条件赋值
这种方式逻辑直观,适合初学者理解:
# 先初始化TrueEarn列为原Earnings的值 df['TrueEarn'] = df['Earnings'] # 针对1991-2005年的数据,重新赋值为Earnings/10000 # 注意:如果Year列是整数类型,用df['Year'].between(1991, 2005);如果是字符串类型,用df['Year'].between('1991', '2005') df.loc[df['Year'].between(1991, 2005), 'TrueEarn'] = df['Earnings'] / 10000
方法二:使用numpy.where函数一行搞定
这种方式更简洁,适合快速实现:
import numpy as np # 同样注意Year列的类型,整数就用1991/2005,字符串就用'1991'/'2005' df['TrueEarn'] = np.where(df['Year'].between(1991, 2005), df['Earnings']/10000, df['Earnings'])
关键提醒
- 先确认你的
Year列数据类型:可以用df.dtypes查看,如果是object类型就是字符串,int64就是整数,条件判断要对应类型,否则会出错。 - 原代码的错误在于用了不符合Python语法的循环和条件写法,Pandas处理这种列赋值不需要手动写for循环,用内置的条件方法就可以高效完成。
内容的提问来源于stack exchange,提问作者Tudor
相关产品推荐
相关产品推荐

