Python数据分析

Python数据分析 知识量:13 - 56 - 232

2.6 选取一维数据><

选取单个元素- 2.6.1 -

数组具有索引,默认从0开始计数,可以通过数组下标来选取数组元素。要选取单个元素,只要标出索引值即可:

import numpy as np
arr=np.arange(10)
print(arr)
print(arr[1])  # 打印第2个元素

运行结果为:

[0 1 2 3 4 5 6 7 8 9]
1

除了正向选取元素外,也可以从数组末尾开始选择,这时下标从-1开始计数:

import numpy as np
arr=np.arange(10)
print(arr)
print(arr[-1])  # 打印最后1个元素

运行结果为:

[0 1 2 3 4 5 6 7 8 9]
9

选取区间元素- 2.6.2 -

如果要选取区间元素,即选取某段元素,可以通过下标指明开始位置和结束位置,下标之间使用冒号(:)分隔。注意:位置默认是左闭右开,即区间包含开始位置的元素,但不包含结束位置的元素。

import numpy as np
arr=np.arange(10)
print(arr)
print(arr[1:5])

运行结果为:

[0 1 2 3 4 5 6 7 8 9]
[1 2 3 4]

可以省略开始位置或结束位置的下标,实现选取某个位置之前或之后的元素:

import numpy as np
arr=np.arange(10)
print(arr)
print(arr[:5])
print(arr[1:])

运行结果为:

[0 1 2 3 4 5 6 7 8 9]
[0 1 2 3 4]
[1 2 3 4 5 6 7 8 9]

正负下标混用选取- 2.6.3 -

可以同时使用正负下标灵活选取元素:

import numpy as np
arr=np.arange(10)
print(arr)
print(arr[1:-1])
print(arr[-8:6])
print(arr[-5:1])  # 空数组

运行结果为:

[0 1 2 3 4 5 6 7 8 9]
[1 2 3 4 5 6 7 8]
[2 3 4 5]
[]

注意:对于实际索引的位置,开始位置不能位于结束位置之后,否则会选不到元素(出现空数组)。

依条件选取- 2.6.4 -

依条件选取元素是Python及NumPy数组的一大特点,提供了高级选择功能,可以通过逻辑表达式来选择某类元素。

import numpy as np
arr=np.arange(10)
print(arr)
print(arr[arr>5])  # 选择大于5的元素

运行结果为:

[0 1 2 3 4 5 6 7 8 9]
[6 7 8 9]