我有一个包含1.000.000行的矩阵,我逐行读取行,并希望在特定列的每个元素上运行一个函数。相关列中的元素是1到1115之间的整数。函数如下:
def vectorized_id(j):
"""Return a 1115-dimensional unit vector with a 1.0 in the j-1'th position
and zeroes elsewhere.
"""
j = j - 1
e = [0] * 1115
e[j] = 1.0
return e
问题是它很慢,如果可能的话,我希望通过优化它来节省时间。可以更快地完成吗?