Как по-разному вращать контур каждого контура matplotlib - PullRequest
0 голосов
/ 06 января 2020

Мне интересно, можно ли повернуть каждый контур контура matplotlib на моем графике под другим углом. Например, у меня есть первый график: contourf plot straight lines, и я хочу повернуть некоторые контуры на основе угла. Я приложил новый эскиз, который я сделал, о том, как я хочу повернуть контуры ниже: contour plot rotated lines, где добавленные линии будут представлять новые контуры. Скажем, контур 0,6-0,8 будет поворачиваться на 55 или -55 градусов от вертикали, контур 0,4-0,6 будет поворачиваться на 45 или -45 градусов от вертикали, а контур 0,2-0,4 будет поворачиваться на 50 или -50 градусов от вертикальный.

Код, который я использовал для создания первого графика, приведен ниже.

import matplotlib.pyplot as plt
import numpy as np

def gaussian(x, mu, sig):
    return np.exp(-np.power(x - mu, 2.) / (2 * np.power(sig, 2.)))
x_values = np.linspace(-5, 5, 120)
y_values = np.linspace(-5, 5, 120)
[xx,yy] = np.meshgrid(x_values,y_values)
xx = np.concatenate(xx)
yy = np.concatenate(yy)

y = gaussian(xx, 0, 1)

plt.figure()
plt.contourf(x_values,y_values,y.reshape(len(x_values),len(y_values)),levels=np.round(np.linspace(0,1,6),decimals=2))
plt.colorbar()
...