[英]3D plot and 3D histogram subplot
这是我的代码:
import numpy as np
import math
from pylab import cm,imshow,colorbar,title,show
import pylab as pyl
from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt
#Parameters
N = 10**2 #Step of discretization
# Cost function
T = np.linspace(0,1,N, False) #Discretization of [0,1]
S = np.linspace(1,2,N,False) #Discretization of [1,2]
X,Y = np.meshgrid(T,S)
C = (X-Y)**2 #Matrix of c[i,j]=(xi-yj)²
def Sinkhorn(M, r, c, lam):
"""
Computes the optimal transport matrix and Slinkhorn distance using the
Sinkhorn-Knopp algorithm
Inputs:
- M : cost matrix (n x m)
- r : vector of marginals (n, )
- c : vector of marginals (m, )
- lam : strength of the entropic regularization
- arret : convergence parameter
Outputs:
- P : optimal transport matrix (n x m)
- dist : Sinkhorn distance
"""
# Uniform measure over [0;1]
uni1 = np.ones(N)
# Uniform measure over [1;2]
uni2 = np.ones(N)
n = 1000
fig = plt.figure()
ax = plt.axes(projection='3d')
# I'm going to compute a matrix which is a approximation of a probability over R^{2}
Gamma_star = Sinkhorn(C, uni1, uni2, 1/10**4)
ax.scatter(X, Y, Gamma_star)
plt.title("Gamma bar 1/{} entre une uniforme([0;1]) et uniforme([1;2])".format(1/10**4))
plt.show()
我的问题:伽玛条收敛到我想调查的一个度量,所以我想打印子图,像这样:(当然它没有用,只是为了告诉你我在想什么)
for i in range(4):
plt.subplot(2,2,i+1)
Gamma_star = Sinkhorn(C, uni1, uni2, 1/10**i)
ax.scatter(X, Y, Gamma_star)
plt.title("Gamma bar 1/{} between uniform([0;1]) and uniform([1;2])".format(1/10**i))
plt.plot()
我还想用 X、Y 和 Z = Gamma_bar 绘制(以完全相同的方式进行子图)3D 直方图,如下所示:
我正在努力,如果有人知道该怎么做,那将是一种解脱,无论如何,感谢您的帮助。
问候。
我设法完成了第一部分(子图),这是我的解决方案:
fig = plt.figure()
for i in range(4):
ax = fig.add_subplot(2, 2, i+1, projection='3d')
Gamma_star, tqui = Sinkhorn(C, uni1, uni2, 1/10**i)
ax.scatter(X, Y, Gamma_star)
plt.title("Gamma bar 1/{} entre une uniforme([0;1]) et uniforme([1;2])".format(1/10**i))
plt.show()
结果:输出
你有什么想法可以改善我的情节吗? 也许添加颜色,也许更改参数,因为所有的图看起来都一样,......欢迎任何帮助:)。
例如,我觉得这个颜色很酷:
无论如何,现在我将专注于 histogram3D 子图。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.