python数组计算cos