Вопросы Теги

Python Matplotlib - Как построить 2D-контур 3D-распределения на одном графике [дубликат]

Решение с использованием пакета sqldf: 

library(sqldf)
sqldf("SELECT Year, Month, COUNT(*) as Freq
       FROM df1
       GROUP BY Year, Month")

4
задан octoback 5 February 2012 в 23:46
поделиться

2 ответа

Вам нужно сгенерировать meshgrid. X, Y и Z должны быть двумерными массивами 

import numpy
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import axes3d

def peaks(x,y):
    return x * numpy.sin(y)

fig = plt.figure()
ax = fig.gca(projection='3d')
X = Y= numpy.arange(-3, 3, 0.1).tolist()
X, Y = numpy.meshgrid(X, Y)

Z = []
for i in range(len(X)):
    Z.append(peaks(X[i],Y[i]))

ax.plot_surface(X, Y, Z, rstride=8, cstride=8, alpha=0.3)
cset = ax.contour(X, Y, Z, zdir='z', offset=-8)
cset = ax.contour(X, Y, Z, zdir='x', offset=-8)
cset = ax.contour(X, Y, Z, zdir='y', offset=8)

ax.set_xlabel('X')
ax.set_xlim(-8, 8)
ax.set_ylabel('Y')
ax.set_ylim(-8, 8)
ax.set_zlabel('Z')
ax.set_zlim(-8, 8)

plt.show()

enter image description here [/g0]

6
ответ дан joaquin 21 August 2018 в 19:08
поделиться
  • 1
    он должен быть запуском, но с этим кодом у меня есть ошибка (с собственной функцией пика, см. выше) TypeError: только массивы length-1 могут быть преобразованы в скаляры Python. Я попробую meshgrid в другом месте – octoback 5 February 2012 в 23:22
  • 2
    numpy.exp () сделал это; благодаря – octoback 5 February 2012 в 23:27

Принимаемый ответ больше не работает. К сожалению, рецензенты отклонили мое предложенное редактирование, которое сделало бы его рабочим. Итак, вот тот же ответ снова, но с небольшими изменениями, необходимыми для его работы в текущей версии matplotlib. 

    import numpy
    import matplotlib.pyplot as plt
    from mpl_toolkits.mplot3d import axes3d

    def peaks(x,y):
        return x * numpy.sin(y)

    fig = plt.figure()
    ax = fig.gca(projection='3d')
    X = Y= numpy.arange(-3, 3, 0.1).tolist()
    X, Y = numpy.meshgrid(X, Y)

    Z = numpy.zeros(X.shape)
    for i in range(len(X)):
        for j in range(len(Y)):
            Z[i,j] = peaks(X[i,j],Y[i,j])

    ax.plot_surface(X, Y, Z, rstride=8, cstride=8, alpha=0.3)
    cset = ax.contour(X, Y, Z, zdir='z', offset=-8)
    cset = ax.contour(X, Y, Z, zdir='x', offset=-8)
    cset = ax.contour(X, Y, Z, zdir='y', offset=8)

    ax.set_xlabel('X')
    ax.set_xlim(-8, 8)
    ax.set_ylabel('Y')
    ax.set_ylim(-8, 8)
    ax.set_zlabel('Z')
    ax.set_zlim(-8, 8)

    plt.show()
1
ответ дан Jonathan Pelham 21 August 2018 в 19:08
поделиться
Другие вопросы по тегам:

Похожие вопросы: