numpy/pandas vectorize custom for loop
问题 I created some example code that mimic what code I got: import numpy as np arr = np.random.random(100) arr2 = np.linspace(0, 1, 20) arr3 = np.zeros(20) # this is the array i want to store the result in for index, num in enumerate(list(arr2)): arr3[index] = np.mean(arr[np.abs(num - arr) < 0.2]) >>> arr3 array([0.10970893, 0.1132479 , 0.14687451, 0.17257954, 0.19401919, 0.23852137, 0.29151448, 0.35715096, 0.43273118, 0.45800796, 0.52940421, 0.60345354, 0.63969432, 0.67656363, 0.72921913, 0