【数据分析&数据挖掘】数组的索引

import numpy as np

# 创建一个一位数组
arr = np.arange(9)
print("arr:
", arr)
# 可以通过下标来获取指定元素
print("获取元素", arr[3])  # 降低维度
print("获取元素", arr[7])  # 降低维度
print("获取元素", arr[[1, 2, 3, 4, 5, 6, 7, ]])
print("获取元素", arr[[1, 3, 5, 7]])

# 通过切片来获取元素
print("切片获取元素:", arr[2:3])
print("切片获取元素:", arr[1:6:2])

# 创建一个二维数组
arr = np.arange(1, 17).reshape((4, 4))
print("arr:
", arr)
print("arr的维度:", arr.ndim)

# 二维数组的索引的时候，使用下标会降低维度，用一次下标降低一个维度
# 使用下标获取二维数组中的元素
# arr[行，列]
print("获取到的元素:", arr[0,1])  # 由二维降到零维
# 使用切片来获取元素
print("获取切片获取到的元素:", arr[0:1, 1:2])  # 保持二维
# 使用下标和切片配合获取元素
print("使用下标和切片获取到的元素:", arr[0,1:2])  # 由二维降到一维

print("~" * 60)

# 二维数组获取多个元素,不降维
print("获取切片获取多个元素的结果: 
", arr[1:3, 1:3])  # 不降维
print("使用下标和切片获取多个元素的结果: 
", arr[[1, 2], 1:3])  # 不降维

# 特殊情况！！
print("使用下标和切片获取多个元素的结果: 
", arr[[1, 2], [1, 3]])  # [ 6 12]
# print("使用下标和切片获取多个元素的结果: 
", arr[[1, 2, 3], [1, 3]])  # 错误，前后的下标列表必须一致

"""
    三维 arr --> 怎么索引元素？ arr[块，行，列]
    四维 arr --> 怎么索引元素？ arr[堆，块，行，列]
    N 维 arr --> 怎么索引元素？ arr[n-1个逗号]     
"""

# 创建一个数组
arr = np.arange(1, 10).reshape((3, 3))
print("arr:
", arr)
print("arr的维度:", arr.ndim)

bool_mask = np.array([0, 1, 2], dtype=np.bool)
bool_mask1 = np.array([0, 1, 2, 3], dtype=np.bool)
print("bool_mask:
", bool_mask)
print("bool_mask:
", bool_mask1)

# 选择True的行，丢掉False的行
print("利用bool数组来进行切片：
", arr[bool_mask, :])
# print("利用bool数组来进行切片：
", arr[bool_mask1, :])  # 错误，bool数组的长度必须与行或列相同
# 选择True的列，丢掉False的列
print("利用bool数组来进行切片：
", arr[:, bool_mask])
# 行列都使用bool数组
print("利用bool数组来进行切片：
", arr[bool_mask, bool_mask])