python - 摆脱没有标题的列

Question

我正在尝试读取一个包含没有标题的列的 CSV 文件。目前，我的解决方案是

df = pd.read_csv("test.csv")
df = df[[col for col in df.columns if 'Unnamed' not in col]]

这看起来有点老套，如果文件中包含带有“未命名”一词的列，则会失败。有更好的方法吗？

Answer 1

read_csv 的usecols 参数函数接受一个可调用函数作为输入。如果您提供的函数对于不需要的列标题计算结果为 False，则这些列将被删除。

func = lambda x: not x.startswith('Unnamed: ')
df = pd.read_csv('test.csv', usecols=func)

我想这个解决方案与您原来的解决方案并没有根本的不同。