Как автоматически извлечь заголовки из файла CSV

Question

мой код проходит через пользовательский файл ввода, если это csv или txt, чтобы найти количество найденных слов, но мне нужно предоставить фиксированный заголовок для него, например, в моем случае: предложения, есть ли способ go через это и пи c автоматически выводит имя столбца из файла, чтобы оно не было фиксированным именем, может быть, в другом файле записаны данные, а не предложения

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

csv examples

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'])

После некоторого обсуждения возможно различить guish между именем столбца (без пробела) и в данных всегда хотя бы один пробел, поэтому общее решение:

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