python - 如何矢量化这个？

Question

我正在寻找对此进行矢量化的方法:

for x in range(1,N+1):
  mul3ou5 = "inf"*(x%3 == 0)+"luans"*(x%5==0) 
  print str(x)*(mul3ou5 =="")+mul3ou5

想法是使用一个 numpy 数组并通过对 numpy 数组的一个操作替换 for 循环 numpy.array(range(100))

提前致谢

Answer 1

虽然@morningsun 的回答很好，但另一个选择是(使用 boolean indexing ):

import numpy as np

x = np.arange(1, N+1)
s = x.astype('S8')
s[x % 3 == 0] = 'inf'
s[x % 5 == 0] = 'luans'
s[x % 15 == 0] = 'influans'

我觉得这更直观，因为它保留了显式模数 (%) 操作。