从pandas中的文本文件为DataFrame创建特性

Question

我希望使用以下代码使用feature_names.txt为数据帧创建功能:

import pandas as pd
features = pd.read_csv("feature_names.txt", header= None)
dataset = pd.read_csv("dataset.csv")
dataset.columns = features.values

输出如下:

(id，)(年龄，)(工人阶级，)(fnlwgt，)(教育，)(教育，)(教育，)(婚姻状况，)(职业，)(关系，)

我希望删除括号和逗号，但使用以下代码:

dataset.columns.str.replace("[()]", "")

这使得所有的头文件都是nan

我将感激任何帮助，非常感谢。

Answer 1

如果'features'是一个系列，您可以尝试:

dataset.columns = features

如果是数据帧，则:

dataset.columns = features['columnName']

或者如果您希望索引值作为列名:

dataset.columns = features.index