第二题其实不用numpy也能过 只要把要计算的位置搞清楚就行