python - 尝试从数据框中删除一些列但抛出名称错误

Question

我已将 CSV 文件读入 Jupyter Notebook，并尝试删除我不会使用的冗余列。我正在使用 drop() 方法，但它给了我一个 NameError!我确信这些列存在，而且我觉得我在这里遗漏了一些明显的东西，但我似乎无法弄清楚。

这是到目前为止我的代码:

  #Calling Libraries
  import os               # File management
  import pandas as pd     # Data frame manipulation
  import numpy as np      # Data frame operations
  import datetime as dt   # Date operations
  import seaborn as sns   # Data Viz

  flight_df=pd.read_csv(r'C:\Users\pc\Desktop\Work\flights.csv')
  # removing na rows
  flight_df.dropna()
  # dropping redundant columns
  newdf=flight_df.drop([O_COUNTRY,O_LATITUDE,O_LONGITUDE,D_COUNTRY,D_LATITUDE,D_LONGITUDE,SCHEDULED_DEPARTURE,DIVERTED,CANCELLED,CANCELLATION_REASON,TAXI_OUT,TAXI_IN,WHEELS_OFF, WHEELS_ON,SCHEDULED_ARRIVAL],axis=1, inplace = True)

抛出此错误:

  NameError                                 Traceback (most recent call last)

~\AppData\Local\Temp/ipykernel_15328/4119461383.py 中2 飞行_df.dropna()3 # 删除冗余列newdf=flight_df.drop([O_COUNTRY,O_LATITUDE,O_LONGITUDE,D_COUNTRY,D_LATITUDE,D_LONGITUDE,SCHEDULED_DEPARTURE,DIVERTED,CANCELLED,CANCELLATION_REASON,TAXI_OUT,TAXI_IN,WHEELS_OFF,WHEELS_ON,SCHEDULED_ARRIVAL],axis=1,inplace = True)

  NameError: name 'O_COUNTRY' is not defined

我尝试定义我想要保留的内容，但它给了我同样的错误

Answer 1

由于本例中的列名称为 str，因此您必须将它们括在 str 分隔符中

newdf=flight_df.drop(['O_COUNTRY','O_LATITUDE','O_LONGITUDE','D_COUNTRY' ...

警告!您正在使用属性 inplace=True 但您尝试将结果分配给新变量。该变量将为None。要么写

flight_df.drop(['O_COUNTRY', ...],axis=1,inplace=True)

或

newdf=flight_df.drop(['O_COUNTRY', ...],axis=1)

与你的 dropna 相同。它不会按照您编写的方式存储。