数据帧中FOR循环中IF语句的性能改进

Question

我在整个数据库的for循环中使用if语句，但处理它花费的时间太长。有没有一种不用for循环的方法？你知道吗

for i in data.index:
    if data['qt_CNAE'][i] > 20:
        data['CNAEuse'][i]=data['CNAE'][i]
        data['CNAEuse_Cres'][i]=data['Crescimento_CNAE'][i]

Answer 1

您应该可以使用2loc语句来完成，大致如下：

data.loc[data['qt_CNAE'] > 20, 'CNAEuse'] = data['CNAE']
data.loc[data['qt_CNAE'] > 20, 'CNAEuse_Cres'] = data['Crescimento_CNAE']

仅供参考，现有的链式索引（例如data['CNAEuse'][i]）在pandas中通常是不好的，通常会导致设置时出现复制错误。最好使用loc运算符或类似的运算符