切片数据帧时的KeyError

Question

我的代码看起来像这样：

d = pd.read_csv('Collector Output.csv')
df = pd.DataFrame(data=d)
dfa = df.copy()
dfa =  dfa.rename(columns={'OBJECTID': 'Object ID', 'test_no': 'TEST #', 'retest_no': 'RETEST #',
                        'tester': 'TESTED BY', 'date': 'DATE', 'proj_no': 'Project Number',
                        'test_elev': 'TEST ELEVATION', 'curve_no': 'CURVE #', 'curve':
                        'Curve Description', 'dry': 'Dry Maximum Density (pcf)', 'opt': 'OPT. M.C.',
                        'Material Type': 'Material Type', 'AC Thickness': 'AC Thickness', 'AB Thickness': 'AB Thickness',
                        'Wet In-place Density, pcf)': 'Wet In-place Density (pcf)', 'inplace_mo': 'FIELD M.C.',
                        'comp_dryde': 'DRY DENSITY', 'relative_c': 'R.C.', 'addl_notes': 'NOTES',
                        'material_t': 'Material Type', 'wet_inpac': 'Wet In-place', 'comp_dryde': 'comp_dryde', 'relative_c': 'relative compaction',
                        'GlobalID': 'GlobalID', 'GlobalID_2': 'GlobalID 2', 'x': 'EAST', 'y': 'NORTH'})


writer = pd.ExcelWriter('Test Test.xlsx', engine= 'xlsxwriter')
dfa.to_excel(writer, sheet_name='Output')
writer.save()

#Slice the dataframe to get the desired columns by their 'name'
dfb = dfa[['TEST #', 'DATE', 'TESTED BY', 'NORTH', 'EAST', 'TEST ELEVATION', 'CURVE #', 'OPT. M.C.', 'FIELD M.C.', 
          'DRY DENSITY', 'R.C.', 'RETEST #', 'NOTES']] 

当我运行它时，我得到一个KeyError，它显示'DRY DENSITY‘'R.C.’。不是索引。我修改了.csv中列的名称是否正确。为什么这两个专栏会有问题？

Answer 1

我想需要双倍的[[]]

cols = ['TEST #', 'DATE', 'TESTED BY', 'NORTH', 'EAST', 'TEST ELEVATION', 'CURVE #', 'OPT. M.C.', 'FIELD M.C.', 'DRY DENSITY', 'R.C.', 'RETEST #', 'NOTES']
dfb = dfa[cols]

如果某些列不匹配，则添加intersection

dfb = dfa[dfa.columns.intersection(cols)]