在循环文件时添加列名

Question

我对 Python 很陌生，所以请多多包涵。我有一个包含 csv 文件的文件夹，其中第一行是我需要处理的数据。所以我需要给他们列名，以便我以后可以调用它们。每个 csv 具有相同数量的列。在我的练习中，我使用了三列。

我了解如何将文件名添加到单个文件：

my_file = pd.read_csv('path\the_file.csv', names = ['first','second','third'])

但我需要转到我的目录并循环浏览大量 csv 文件。老实说，我什至不知道该怎么做（我知道很难过）。我已经设法使用 os.listdir 遍历文件名，但是当我需要其中的数据时，这对我没有多大用处。一旦我得到这些列名，我就知道该怎么做了。

尽可能使用 pandas 是非常可取的。我看了很多，但似乎找不到任何真正有效的东西。我真的很感激帮助！

编辑： 这是我将要做的一部分，但需要为文件夹中的所有 csv 文件做。

my_file = pd.read_csv('path\the_file.csv', names=['first','second','third'])
first_col = my_file['first']
second_col = my_file['second']
third_col = my_file['third']
key_codes = []
key_codes.append(second.map(str) + third.map(str))

所以，如果第 2 列有“123”，第 3 列有“4”，那么我正在制作“1234”，但现在我只需要弄清楚如何循环文件并为它们添加相同的名称/标题。

Answer 1

你需要IIUCglob:

#glob can use path with *.txt - see http://stackoverflow.com/a/3215392/2901002
import glob
key_codes = []
for files in glob.glob('files/*.csv'):
    df = pd.read_csv(files, names = ['first','second','third'])
    key_codes.append(df.second.map(str) + df.third.map(str))

通过iloc选择second和third列的另一种解决方案，还删除了参数names并将header=None添加到read_csv：

#glob can use path with *.txt - see http://stackoverflow.com/a/3215392/2901002
import glob
key_codes = []
for files in glob.glob('files/*.csv'):
    df = pd.read_csv(files, header=None)
    key_codes.append(df.iloc[0,1].astype(str) + df.iloc[0,2].astype(str))