Python -在二进制数组中查找一维质心的最佳方法

import numpy as np

def find_centroid(arr):
    total_sum = np.sum(arr)
    cumulative_sum = np.cumsum(arr)
    centroid_index = np.where(cumulative_sum >= total_sum / 2)[0][0]
    centroid_value = centroid_index + 1 if cumulative_sum[centroid_index] == total_sum / 2 else centroid_index
    return centroid_value

binary_array = np.array([0, 1, 1, 0, 1, 0, 0, 1, 1, 1])

centroid = find_centroid(binary_array)
print("一维质心的索引位置为:", centroid)