NumPy数组元素获取：a[i][j]与a[i,j]哪种方法更优？

两者的本质区别 #

• a[i][j]是两次独立索引：先提取数组的第i行（返回一个一维子数组），再从这个子数组中取第j个元素。
• a[i,j]是一次多维索引：直接通过元组(i,j)定位二维数组中的目标元素，属于NumPy原生支持的多维数组索引语法。

性能对比 #

a[i,j]的性能明显更优，因为它不需要生成中间子数组，减少了内存分配和函数调用的额外开销。可以通过简单的性能测试验证：

import numpy as np
import timeit

# 创建一个大的二维数组
a = np.arange(1_000_000).reshape(1000, 1000)

# 测试两种索引方式的耗时
time_ij = timeit.timeit(lambda: a[500][500], number=100_000)
time_comma = timeit.timeit(lambda: a[500, 500], number=100_000)

print(f"a[i][j] 耗时: {time_ij:.4f} 秒")
print(f"a[i,j]   耗时: {time_comma:.4f} 秒")

运行后会看到a[i,j]的耗时通常只有a[i][j]的一半甚至更少。

可读性与扩展性 #

在处理高维数组时，a[i,j,k]的写法比a[i][j][k]更简洁直观，符合NumPy多维数组的设计逻辑，也能降低代码出错的概率。

总结推荐 #

优先使用逗号分隔的多维索引方式a[i,j]，它在性能、可读性和扩展性上都全面优于a[i][j]的写法。

内容的提问来源于stack exchange，提问作者darren