フォルダ内のさまざまな種類のcsvファイルの列見出しを検索するにはどうすればよいですか？

Question

私は、さまざまな列の場所で異なるファイルを摂取する必要があるという問題があります。 1つのファイル列は4行下に開始し、別のファイル列は1行目で開始する場合があります。

1つのファイルは次のようになります。

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

もう一つは、1行目に、このような列があるとします。

column 1, column 2, column 3 

は、私はすべてのファイルの列ヘッダーのリストを取得する必要があります。私は、列見出しを3つ以上の項目と見なします。もし私がcsvモジュールを使用しているなら、これをどのように書くことができますか？

私のようなものがあります。それはまた、1つの文字列を含む列を返しているので、これは非常に機能していない

temprow = next(csvfile) 

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

を。

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

は、このコードは動作します。 4つ以上の要素（3つ以上の要素）を持つ.csvファイルの最初の行を返します。

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)