如何從 csv 文件中自動獲取標題

Question

我的代碼正在通過用戶輸入文件，如果它是一個 csv 或一個 txt 來查找找到的單詞數，但我需要為它提供一個固定的標題，例如在我的情況下：句子，有沒有辦法通過這個和圖片自動從文件中提取列名，因此它不會是固定名稱，也許另一個文件會寫入數據而不是句子

wdata=pd.read_csv(fileinput)
nowt=wdata['sentences'].str.split().map(len).sum()

csv 示例

Answer 1

如果只有一列先通過指定新的列名names參數，然后通過跳過原來的CSV標題skiprows參數read_csv ：

wdata=pd.read_csv(fileinput, names=['col name'], skiprows=1)

如果沒有 csv 標頭：

wdata=pd.read_csv(fileinput, names=['col name'])

經過一些討論，可以區分列名（那里沒有空格）和數據中總是至少有一個空格，所以一般的解決方案是：

header = pd.read_csv(file, nrows=0).columns[0] 
skip = int(header.count(' ') == 0) 
df = pd.read_csv(file, names=['col'], skiprows=skip)