如何在python中选择特定索引的元素？

Question

说，

test_data = pd.read_csv('test/path'/.csv')
train_data = pd.read_csv('train/path'/.csv')
example_data = pd.read_csv('example/path'/.csv')
more_data = pd.read_csv('more/path'/.csv')

data_list = [test_data, train_data, example_data, more_data]

for i in data_list:
    i.rename(columns = lambda x: x.replace('data','df'), inplace=True)

我想要获得的是A = [1,2,3,4] 的具体元素，例如只有第一和第三个元素。

所以输出应该如下所示：

A

我该怎么做？

[1, 3] 不是我所知道的正确命令，所以寻找正确的命令。

Answer 1

您可以尝试使用slice notation。

print(A[::2]) # this will print each second element of the list.

Answer 2

你可以做一点点理解：

代码：

[A[i] for i in (0, 2)]

测试代码：

A = [1, 2, 3, 4]
print([A[i] for i in (0, 2)])

结果：

[1, 3]

Answer 3

试试这个。这适用于您放入下面较小列表的任何索引。

Response 401 Unauthorized