9、where函数和take函数

where函数会根据指定条件返回所有满足条件的数组元素的索引值。where函数返回的是多维数组,如果需要提取其中的某个元素时,需要先用np.ravel将其展平后提取。即秩为2。

take函数可依照索引值从数组中取出相应的元素。

图1 实例

举例进行VWAP和TWAP的重新计算:

图2 TWAP的示例

你可能感兴趣的:(9、where函数和take函数)