从[pandas.core.frame.DataFrame]变量中提取特定数据
从[pandas.core.frame.DataFrame]变量中提取特定数据
提问于 2020-07-24 01:50:03
在使用pandas从.csv文件中提取数据时，我希望收集该文件中各种列的标签。而不是硬编码，我试图从我从以下代码创建的变量中提取它：
train_data = pd.read_csv("Anydatasheet.csv")
features = ["Pclass","Age", "Fare", "Parch", "SibSp","Sex","Embarked"]
X = pd.get_dummies(train_data[features])
X.head()(上面的标签是指所附图片中圈出的粗体文字)
有人能告诉我怎么做吗？
(图片数据来源: Kaggle titanic data)
回答 1
Stack Overflow用户
回答已采纳
发布于 2020-07-24 16:51:53
您要查找的是列名。你可以直接阅读它们：
train_data = pd.read_csv("Anydatasheet.csv")
features_name = train_data.columns或者，如果您想要它们作为python常规列表：
train_data = pd.read_csv("Anydatasheet.csv")
features_name = train_data.columns.tolist()示例：
import pandas as pd
df = pd.DataFrame({"city":[1,1], "A":[5,6]})
print(df.columns.tolist())输出：
['city', 'A']页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/63060259
复制相关文章
相似问题
腾讯云开发者
