Корреляция радиального расстояния до 2D mgrid на python

Question

У меня есть два массива 1D. Один из которых содержит температуру и другое радиальное расстояние (для каждой соответствующей температуры). Я хочу создать график типа карты тепла, используя эту информацию. Вот где я столкнулся с проблемами: 1. Если я создаю сетку с 2d-сетками, как мне сопоставить радиальное расстояние до каждого? Скажем, радиальное расстояние составляет 5 единиц, как найти все квадраты сетки, которые находятся на расстоянии 5 единиц от центра? 2. Затем, как я сопоставляю каждой температуре свой соответствующий набор точек сетки. Так скажите, что умеренный составляет 20 градусов на радиальном расстоянии 5, как я могу выразить это, поскольку он равен 20 градусам при следующем наборе x, y квадратов сетки?

Спасибо за любую помощь.

Answer 1

meshgrid - ваш друг здесь. Сначала настроить сетку плюс x и y координат сеток (вы будете иметь два 5 от 5 массивов):

import numpy as np 
x, y = np.meshgrid(np.arange(-2, 3), np.arange(-2, 3)) 
heatmap = 0 * x # easy way to get shape right 

Теперь, поддельные некоторые данные:

r = np.array((0, 0.5, 1.5, 2.5)) # Your radial distance 
T = np.array((100, 90, 70, 40)) # Your temperature at distance 

Overlay данные с внутренней стороны наружу, начиная с середины (при условии, r монотонно возрастает):

r2 = r**2 
xy2 = x**2 + y**2 
for ii in range(r.size): 
    heatmap[np.where(xy2 >= r2[ii])] = T[ii] 

Вот и все. Вот результирующая карта тепла:

array([[ 40, 70, 70, 70, 40], 
     [ 70, 90, 90, 90, 70], 
     [ 70, 90, 100, 90, 70], 
     [ 70, 90, 90, 90, 70], 
     [ 40, 70, 70, 70, 40]])