首页 > 解决方案 > 使用 pandas 连接 2 个文本文件,第一个文本文件到标题,第二个作为正文

问题描述

我正在使用 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 网站寻求帮助,但我仍然不明白。我被告知这足以解决这个问题。

标签: pythonpandasjupyter

解决方案


问题是函数的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)

推荐阅读