Тепловая карта с использованием Matplotlib с данными в трех столбцах CSV - PullRequest
0 голосов
/ 10 марта 2019

Используя Python matplotlib, мне нужно построить тепловую карту из файла CSV, где первый и второй столбцы имеют переменные x и y соответственно.И третий столбец имеет переменную z (которая является функцией от x и y).Я хотел бы представить переменную z в ​​цвете, используя двухмерную тепловую карту.

Переменные x и y в файле csv расположены не в определенном порядке.

Я начал работать над некоторым кодом для создания матрицы из файла, но не понял, как поступить.

def heat_map_simple(csv_file,x_var,y_var, color_var):

    data = pd.read_csv(csv_file)
    data = pd.DataFrame(data)
    x_n = len(set(data.iloc[:][x_var]))
    y_n = len(set(data.iloc[:][y_var]))
    matrix = np.zeros((x_n, y_n))
    rows = data.shape[1]
    for r in xrange(rows):
        val = data.iloc[r][color_var]
...