假设有一个列表[0,1,2,3,4,5,6,7,8,9]我想取在3.5距离最小的两个值输出
收起
熟悉numpy的话,可以这样写:
>>> import numpy as np >>> data = np.array([0,1,2,3,4,5,6,7,8,9]) >>> data[np.argsort(np.abs(data-3.5))[:2]] array([3, 4])
报告相同问题?