在pandas的列上应用lambda

时间:2017-01-25 09:58:29

标签: python pandas lambda pandas-apply

我有以下数据框

ipdb> csv_data
  country_edited  sale_edited  date_edited  transformation_edited
0          India       403171     21091956                      1
1         Bhutan       394096     21091956                      2
2          Nepal       361372     21091956                      3
3         madhya       355883     21091956                      4
4          sudan       262949     21091956                      5

以下是我的代码

transfactor_count = 5.6
csv_data["transformation_edited"] = csv_data["transformation_edited"].apply(lambda x: x * transfactor_count)

但是上面的代码给了我一个错误

*** NameError: global name 'transfactor_count' is not defined

如何解决这个问题?

实际代码

for foreign_key in data_mapping_record.csvdatabasecolumnmapping_set.all():
    data_type = foreign_key.data_type
    trans_factor = foreign_key.tranformation_factor
    if data_type == "Decimal":
        import ipdb; ipdb.set_trace()
        csv_data[foreign_key.table_column_name] = csv_data[foreign_key.table_column_name].apply(lambda x: x * trans_factor )
    elif data_type in ["Date", "Datetime"]:
        csv_data[foreign_key.table_column_name] = csv_data[foreign_key.table_column_name].apply( lambda d: datetime.strptime(d, dates[date]) )

2 个答案:

答案 0 :(得分:2)

正如错误已经说明的那样,lambda函数无法找到全局变量。 您可以尝试通过以下方式使其全球化:

global transfactor_count
transfactor_count = 5.6
csv_data["transformation_edited"] = csv_data["transformation_edited"].apply(lambda x: x * transfactor_count)

但正如以色列指出:

csv_data["transformation_edited"] = csv_data["transformation_edited"] * transfactor_count 

更优雅。

您可能需要在以下情况下将数据类型更改为float:

csv_data["transformation_edited"] = csv_data["transformation_edited"].astype(float) * transfactor_count

答案 1 :(得分:-1)

您可以在数据框架上使用lambda,如下所示:

transfactor_count = 5.6;
csv_data['transformation_edited']=map(lambda x: x * transfactor_count, csv_data['transformation_edited'])