python - 结合iloc和loc

我正在尝试将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]进行切片索引

有没有办法做到这一点？

非常感谢

最佳答案

您可以链接此操作，或仅将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

关于python - 结合iloc和loc，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/55435469/