使用下面的代码选择列，但仅获得顶行

Question

使用熊猫选择列，但仅获得第一行。使用以下代码

data1=data.loc[:,'Subject':'Sub-Component']

#column names being "Subject", "Original Product Version","Software Version",   "Software Version #","Software Release","Component","Sub-Component"

data1=data.loc[:,'Subject':'Sub-Component']

我希望这会选择所有列，但不会选择所有列，而我只会得到结果的行名/标题

Answer 1

如果要选择特定的列，请将这些列的名称添加到列表中并传递该名称：

col_list = ['Subject', 'Original Sub-Component']
data1 = data.loc[:, col_list]

如果要选择列的范围，请改用iloc：

data1 = data.iloc[:, 3:8]