在csv文件中,若存在负数,则把该负数所在的列删除。
import pandas as pd
df_dir = 'df.csv' # 文件名
df = pd.read_csv(df_dir) # 读取csv文件
for j in range (0,len(df.index)): # 遍历每行
# 查找一行中是否有数据小于0,若有,返回该数据所在的列名
cols=[x for i,x in enumerate(df.columns) if float(df.iat[j,i]) < 0]
df=df.drop(cols,axis=1) # 用.drop()删除带有负数的列
df.to_csv('df_new.csv') # 保存csv文件