要按照自定义的顺序将np.ndarray展平,可以使用numpy的nditer函数来迭代数组,并按照指定的顺序输出数组的元素。以下是一个示例代码:
import numpy as np
# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 自定义的展平顺序,按照从右下角到左上角的顺序展平
order = 'F'
# 使用nditer函数迭代数组
for x in np.nditer(arr, order=order):
print(x, end=' ')
输出结果为:
9 6 3 8 5 2 7 4 1
在上述示例中,我们创建了一个2x2的数组arr,并使用nditer函数按照自定义的顺序(order='F',从右下角到左上角)展平数组。在循环迭代中,我们打印每个元素的值。
下一篇:按照自定义分类顺序删除行