【Numpy】二维数组的切片matrix[0, :]和索引matrix[[0],:]

• 提醒：结合chatgpt生成代码分析，不想看可以不看或者自己去问chatgpt
• 比较抽象的概念，建议自己多上手写一下代码分析一下
• 以下讨论是二维数组，三维数组建议自己再分析一下，随机数生成数组数据不一样，主要对比原数组，注意输出向量维度
  

  文章目录

  • 二维数组选择一行matrix[0,:]或者一列matrix[:,0]
  • matrix[0, :]结合数组切片[0:2]操作取出前两个元素matrix[0, 0:2]
  • 使用索引进行选择matrix[[0], :]
  • 索引matrix[[0, 2], :]和切边matrix[0:2, :]

二维数组选择一行matrix[0,:]或者一列matrix[:,0]

• 首先创建一个包含随机选择的0、1、2元素的大小为5x3的矩阵

matrix = np.random.choice([0, 1, 2], size=(5, 3))

:代表这一行的数据全部选择
选择一行输出为[0 0 2]，形状为(3,)
matrix[0, :] 选择二维数组的第一行。
matrix[1, :] 选择二维数组的第二行。
matrix[-2, :] 选择二维数组的倒数第二行。
matrix[-1, :] 选择二维数组的最后一行。

import numpy as np

# 创建一个包含随机选择的0、1、2元素的大小为5x3的矩阵
matrix = np.random.choice([0, 1, 2], size=(5, 3))

# 输出结果
print("Random matrix:")
print(matrix)

# 选择第一行和第二行
first_row = matrix[0, :]
second_row = matrix[1, :]

print(f"matrix[0,:] shape {
     
     matrix