Соединение 2 текстовых файлов с использованием панд, 1-й текстовый файл в заголовок, 2-й в качестве тела

Question

Я использую Jupyter, и у меня есть 2 текстовых файла.dataset.txt и feature_names.txt.Я ввел следующий код.

header1 = r'./data/feature_names.txt'
main = r'./data/dataset.txt'
df = pd.read_csv(main, names=[header1])

Вывод

По какой-то причине он печатается только ./data/feature_names.txt, хотя при печати заголовка он печатается нормально.Но единственный вопрос, когда я присоединяюсь к нему с основным.Я не уверен, как сделать заголовочный файл заголовком.Я пользуюсь Jupyter Website для помощи, но до сих пор не понимаю.Мне сказали, что этого будет достаточно, чтобы решить эту проблему.

Answer 1

header1 = r'./data/feature_names.txt' #header1 header2 header3
with open(header1,'r') as file:
    header_values = file.read().split() # you need to read the headers from file
main = r'./data/dataset.txt'
df = pd.read_csv(main, names=header_values)

Answer 2

Проблема в том, что параметр names функции read_csv ожидает массив имен.Вместо этого вы передали ему имя файла, который содержит имена ваших столбцов.Попробуйте это:

header1 = r'./data/feature_names.txt'
header_file = open(header1, 'r')

# Assuming one column name per line
headers = []
for line in header_file:
    headers.append(line.strip())
header_file.close()

main = r'./data/dataset.txt'
df = pd.read_csv(main, names=headers)