在python中打印2列

Question

df_heartdis = pd.read_csv('processed.cleveland.data.csv', na_values='?')
df_heartdis.columns['age', 'sex', 'cp', 'trestbps', 'chol', 'fbs', 'restecg', 'thalach', 'exang', 'oldpeak', 'slope',\
    'ca', 'thal', 'num']

print('class labels = {}'.format(df_heartdis(['num'],['chol'])))

在我的数据框中，我有14列，其中，我想并排打印2列。我尝试使用

print('class labels = {}'.format(df_heartdis(['num'],['chol']))) 

但是它会抛出错误，但可以使用

打印一列

print('class labels = {}'.format(df_heartdis(['num'])))

Answer 1

在创建数据框时提供列标题。 使用.loc函数选择特定列。

下面的代码应该有效：

df_heartdis = pd.read_csv('processed.cleveland.data.csv', na_values='?', names = ['age', 'sex', 'cp', 'trestbps', 'chol', 'fbs', 'restecg', 'thalach','exang', 'oldpeak', 'slope','ca', 'thal', 'num'])

print('class labels = {}'.format(df_heartdis.loc[:, ['num', 'chol']]))

此处，.loc函数会选择:和num列的所有行（chol）。