开始用numpy找索引，后来发现不能用numpy然后自己些，快写完时间没了