loc与loc的结合

Question

我正在尝试把iloc和loc结合起来，有可能吗？

具体来说，我想：

• 在表单中为行值提供整数(:train_size)
• 给出列值的列名列表(替换下面代码中的0,1 ) training_set = dataset.iloc[:train_size,0,1].values

正在尝试

training_set = dataset.loc[:train_size,[list_input_and_y_parameters]].values

给出错误消息

TypeError:无法使用这些索引器4275进行切片索引

有办法吗？

非常感谢

Answer 1

对于列表中列的位置，可以链接此操作，或仅将iloc与Index.get_indexer一起使用：

training_set = dataset.iloc[:train_size].loc[:, ['col1','col2']].values

training_set = dataset.iloc[:train_size, df.columns.get_indexer(['col1','col2'])].values

Answer 2

由于链接loc和iloc可能导致SettingWithCopyWarning，因此可以使用不需要使用Index.get_indexer的选项(假设索引中没有重复项)：

training_set = dataset.loc[dataset.index[:train_size], ['col1','col2']].values