如何在文件夹中找到各种类型的csv文件的列标题？

Question

我有一个问题，我需要使用不同的列位置接收不同的文件。一个文件列可能开始向下 4 行，而另一个文件列可能从第一行开始。

一个文件可能如下所示：

This
is
a
column 1, column 2, column 3, column 4

另一个可能在第 1 行有这样的列：

column 1, column 2, column 3

我需要获取每个文件列标题的列表。我认为列标题是大于 3 项的列表。如果我使用的是 csv 模块，我该怎么写？

我有类似的东西：

temprow = next(csvfile)

for value in temprow:
    if value == '':
        temprow = next(csvfile)
    if len(value) > 3:
        header = temprow
    else:
        header = temprow

这不太有效，因为它还返回包含 1 个字符串的列。

Answer 1

试试这个：

with open('yourfile.csv', 'r') as f:
    for line in f:                        # iterate for each line
        if "," in line:                   # the header line should contain comma     
            header = line          
            break                         # break the loop when header line is found

print(header)

输出：

column 1, column 2, column 3, column 4

Answer 2

根据您帖子中的规范，此代码有效。它返回 .csv 文件中包含 4 个或更多元素（“大于 3 个项目”）的第一行。

headers = []                                        # Column names will be appended to this list
files = ['./test']                                  # Insert files here

for f in files:                                     # Loop over files
    with open(f, 'r') as fh:                        # Open file
        reader = csv.reader(fh, delimiter = ',')    # Create reader
        for row in reader:                          # Loop over rows
            if len(row) >= 4:                       # Criteria for appending to headers
                headers.append(row)