python - Pandas 中具有多个 IF 条件的 For 循环

Question

我想用 panda DataFrame 中的列的条件进行 for 循环:

import numpy as np  
import pandas as pd


df=pd.DataFrame(pd.read_csv("data.csv"))  
print df  

DWWC1980     DWWC1985   DWWC1990  
16.7140310  16.35661439 15.89201716  
20.9414479  18.00822799 15.73516051  
33.95022337 51.87065104 73.76376497  
144.7000805 136.1462017 130.9143924  
54.9506033  75.03339188 93.22994974  

For循环条件语句:

for i in range (1980,2015,5):

    if   any(df["DWWC"+str(i)] <=18.25)  :

            df['MWTP'+str(i)]=(((10-33)/(5))*(df["DWWC"+str(i)]-5))+10  

    elif any((df["DWWC"+str(i)] >  18.25) &  (df["DWWC"+str(i)] <= 36.5)) :

            df['MWTP'+str(i)]=((10/(df.two-df.three))*(df["DWWC"+str(i)]-df.three))+df.Three

    else :
            df['MWTP'+str(i)]=(((df.Three_value-6)/(df.three-5))*(df["DWWC"+str(i)]-6  

df.to_csv('MWTP1.csv',index='ISO3')

但是当我运行这段代码并与手动计算进行比较时，我发现只有第一个条件计算是正确的，而其他条件则不成立。 (df.one、df.two 和 df. Three 是其他列。)

  MWTP1980       MWTP1985         MWTP1990  
 25.87096095    30.72758886  37.04060109  
 -77.06996017   20.00112954      95.22533503  
 -290.1012655   -640.6304196    -1068.866556  
 -1845.172654   -1718.865351    -1641.61201  
 -1397.638671   -2171.737373    -2873.130596  

Answer 1

您可以使用numpy.select对于获取列名称格式:

for i in range (1980,2015,5):
    m1 = df["DWWC{}".format(i)] <=18.25
    #inverted m1 mask by ~
    m2 = ~m1 & (df["DWWC{}".format(i)] <= 36.5)
    a = (((10-33)/(5))*(df["DWWC{}".format(i)]-5))+10 
    b = ((10/(df.two-df.three))*(df["DWWC{}".format(i)]-df.three))+df.Three
    c = (((df.Three_value-6)/(df.three-5))*(df["DWWC{}".format(i)]-6

    df["MWTP{}".format(i)] = np.select([m1,m2],[a,b], default=c)