将colorbar离散化

本文介绍如何在Python中对colorbar进行离散化处理,以更好地展示数据的分段和层次。通过控制colorbar的刻度和颜色间隔,可以使得可视化结果更加清晰易读。
摘要由CSDN通过智能技术生成
from scipy.io import loadmat
import matplotlib
import matplotlib.pyplot as plt
plt.rc('font', family="Times New Roman")  # 设置整个
import pandas as pd
import numpy as np
matplotlib.rcParams['xtick.direction'] = 'in'
matplotlib.rcParams['ytick.direction'] = 'in'

# 读数据
mat_path = r'C:\Users\Dell\Desktop\Cloud Model\result mat\aresprcp.mat'
recieve = loadmat(mat_path)
temp = recieve['resprcp']
temp = temp.T
miuset = pd.DataFrame(temp)
miuset.columns = np.arange(1,73)    # 设置列索引
miuset.index = ['1963-2002','1964-2003','1965-2004','1966-2005','1967-2006','1968-2007','1969-2008','1970-2009',
                '1971-2010','1972-2011','1973-2012','1974-2013','1975-2014','1976-2015','1977-2016','1978-2017'] # 设置行索引

# 画图
f, ax1 = plt
import numpy as np import matplotlib.pyplot as plt # 光纤参数 core_radius = 5e-6 # 光纤芯径 cladding_radius = 125e-6 # 包层芯径 n_core = 1.45 # 光纤芯的折射率 n_cladding = 1.44 # 包层的折射率 alpha = 0.2 # 损耗系数 # 模式参数 m = 1 # 模式数 l = 0 # 角动量数 k = 2 * np.pi / 1.55e-6 # 波矢量 # 离散化 dr = 1e-7 # 径向离散化步长 dz = 1e-5 # 纵向离散化步长 r_max = 2 * core_radius # 最大径向范围 z_max = 1e-3 # 最大纵向范围 nr = int(r_max / dr) + 1 # 径向离散化数 nz = int(z_max / dz) + 1 # 纵向离散化数 # 初始化 r = np.linspace(0, r_max, nr) z = np.linspace(0, z_max, nz) E = np.zeros((nr, nz), dtype=complex) # 边界条件 E[:, 0] = np.exp(1j * k * r) # 入射光线 E[:, -1] = 0 # 输出面边界条件 # 模式初值 w = np.sqrt(2 / np.pi) * np.exp(-r ** 2 / core_radius ** 2) w *= np.sqrt((2 * l + 1) / (2 * np.pi * m * core_radius ** 2)) w /= np.sqrt(np.sum(np.abs(w) ** 2) * dr) E[:, 1] = w # 数值求解 for i in range(1, nz - 1): # 径向二阶导数 d2Edr2 = (E[2:, i] - 2 * E[1:-1, i] + E[:-2, i]) / dr ** 2 # 纵向一阶导数 dEdz = (E[:, i + 1] - E[:, i]) / dz # 光学传输方程 E[1:-1, i + 1] = E[1:-1, i] + dz * ( (1j * k * n_core) ** 2 * E[1:-1, i] - (1 / core_radius ** 2 + alpha / 2) * E[1:-1, i] - ( n_core ** 2 - n_cladding ** 2) * d2Edr2 / k ** 2 - 2 * 1j * k * dEdz / (m * core_radius ** 2)) # 绘图 plt.imshow(np.abs(E) ** 2, extent=(0, z_max, r_max, 0), aspect='auto') plt.xlabel('z / m') plt.ylabel('r / m') plt.colorbar() plt.show()
05-29
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

大雨海深

感谢您的支持和鼓励

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值