python学习笔记

1、np.ndenumerate
作用：用于在多维数组上进行迭代。这个函数返回一个迭代器，它生成一个包含数组索引和对应元素值的元组。
用法：

import numpy as np

arr = np.array([[1, 2], [3, 4]])
for index, value in np.ndenumerate(arr):
    print(index, value)

结果：

(0, 0) 1
(0, 1) 2
(1, 0) 3
(1, 1) 4

2、np.newaxis
作用：用于增加数组的维度。在数组中使用 np.newaxis 可以在指定位置增加一个新的轴，这通常用于改变数组的形状而不需要复制数据。
用法：
将一维数组转换为二维数组的行向量：

import numpy as np

arr = np.array([1, 2, 3])
arr_2d = arr[np.newaxis, :]
print(arr_2d)
# 输出：
# [[1 2 3]]

3、np.repeat
可以沿着指定的轴重复数组中的元素多次，生成一个新的数组。
语法：numpy.repeat(a, repeats, axis=None)

import numpy as np

# 创建一个简单的数组
a = np.array([1, 2, 3])

# 重复数组中的每个元素两次
repeated_array = np.repeat(a, 2)
print(repeated_array)  # 输出: [1 1 2 2 3 3]

# 创建一个二维数组
a_2d = np.array([[1, 2], [3, 4]])

# 沿着水平轴（axis=1）重复每个元素两次
repeated_2d_array = np.repeat(a_2d, 2, axis=1)
print(repeated_2d_array)
# 输出:
# [[1 1 2 2]
#  [3 3 4 4]]

# 沿着垂直轴（axis=0）重复每行两次
repeated_2d_array_axis0 = np.repeat(a_2d, 2, axis=0)
print(repeated_2d_array_axis0)
# 输出:
# [[1 2]
#  [1 2]
#  [3 4]
#  [3 4]]

原文地址：https://blog.csdn.net/qq_32649321/article/details/142368070

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！