Получение индекса некоторого элемента в массиве Numpy 2D

Question

Я новичок в Python, так что я нужна помощь с этим:

У меня есть 2D массив чисел, представляющих плотность окружности материала в пространстве и Я хочу найти центр. Итак, я хочу получить индекс чисел, представляющих диаметр, а затем средний индекс будет центром. В этом коде я сохраняю только значение плотности: tempdiameter.append (cell) Я хочу индекс самой ячейки. Как мне это сделать. Также я не хочу использовать списки для диаметра. так как я могу создать динамический массив 1D np? Благодаря

for row in x: 
    for cell in row: 
     if cell!=0: 
      tempdensity+=cell 
      tempdiameter.append(cell) 

    if tempdensity>maxdensity: 
      maxdensity=tempdensity 
      if len(tempdiameter)>=len(diameter): 
        diameter=tempdiameter 

    tempdensity=0 
    tempdiameter=[] 

Answer 1

Чтобы получить строку с наибольшим числом ненулевых ячеек и самой высокой суммой вы можете сделать

densities = x.sum(axis=1) 
lengths = (x > 0).sum(axis=1) 
center = x[(densities == densities.max()) & (lengths == lengths.max()] 

Try, чтобы избежать использования петель в numpy. Дайте мне знать, если вы этого не хотите, и я постараюсь ответить на него лучше. При задании вопроса вы должны указать ввод/вывод образца. Вы также можете изменить свой вопрос, а не добавлять комментарии.