是否有任何方式在numpy中获得对数组对角线的引用? 我希望我的阵列对角线按一定因素分 感谢Numpy修改ndarray对角线
10
A
回答
16
14
一个快速的方法来访问对角方形(n,n)
numpy的数组是arr.flat[::n+1]
:
n = 1000
c = 20
a = np.random.rand(n,n)
a[np.diag_indices_from(a)] /= C# 119 microseconds
a.flat[::n+1] /= C# 25.3 microseconds
4
的np.fill_diagonal
功能是相当快的:
np.fill_diagonal(a, a.diagonal()/c)
其中a
是您的阵列和c
是你的因素。在我的机器上,这种方法与@ kwgoodman的a.flat[::n+1] /= c
方法一样快,在我看来有点清晰(但不是光滑的)。
3
比较上述3种方法:
import numpy as np
import timeit
n = 1000
c = 20
a = np.random.rand(n,n)
a1 = a.copy()
a2 = a.copy()
a3 = a.copy()
t1 = np.zeros(1000)
t2 = np.zeros(1000)
t3 = np.zeros(1000)
for i in range(1000):
start = timeit.default_timer()
a1[np.diag_indices_from(a1)] /= c
stop = timeit.default_timer()
t1[i] = start-stop
start = timeit.default_timer()
a2.flat[::n+1] /= c
stop = timeit.default_timer()
t2[i] = start-stop
start = timeit.default_timer()
np.fill_diagonal(a3,a3.diagonal()/c)
stop = timeit.default_timer()
t3[i] = start-stop
print([t1.mean(), t1.std()])
print([t2.mean(), t2.std()])
print([t3.mean(), t3.std()])
[-4.5693619907979154e-05, 9.3142851395411316e-06]
[-2.338075107036275e-05, 6.7119609571872443e-06]
[-2.3731951987429056e-05, 8.0455946813059586e-06]
所以你可以看到,np.flat
方法是最快的,但是很小。当我再次运行这几次时,有时候fill_diagonal
方法稍微快一点。但可读性明智的可能值得使用fill_diagonal方法。
相关问题
- 1. 在Python中获取角落值numpy ndarray
- 2. 在numpy的ndarray
- 3. Numpy NdArray Memoization
- 4. 内numpy的ndarray
- 5. 在Python(numpy)中修改ndarray的正确方法是什么?
- 6. 如何沿轴线平坦numpy ndarray?
- 7. numpy数组到ndarray
- 8. numpy的ndarray乘法
- 9. subclassing numpy ndarray问题
- 10. 的Python NumPy的ndarray
- 11. NumPy ndarray.all()与np.all(ndarray)vs all(ndarray)
- 12. 逆转numpy的蟒蛇对角线
- 13. 在numpy中对角线堆叠?
- 14. 获取对角线而不使用numpy?
- 15. 2D numpy数组的上对角线
- 16. 如何从numpy ndarrays创建numpy ndarray?
- 17. 在numpy中更改矩阵对角线的值
- 18. 修改角阵对象onload
- 19. numpy ndarray切片和迭代
- 20. 替换numpy ndarray中的值
- 21. Numpy ndarray与阵列切片
- 22. numpy ndarray高级索引
- 23. 在numpy的ndarray(Python)的
- 24. 正常化numpy ndarray数据
- 25. 我如何从numpy的ndarray
- 26. 最大奇异值NumPy`ndarray`
- 27. Numpy/Scipy:如何重建ndarray?
- 28. numpy ndarray的查找索引
- 29. 从给定的numpy数组创建块对角线numpy阵列
- 30. numpy中是否有函数替换numpy数组的低对角线值和高对角线值?