本文整理汇总了Python中xarray.DataArray.sum方法的典型用法代码示例。如果您正苦于以下问题:Python DataArray.sum方法的具体用法?Python DataArray.sum怎么用?Python DataArray.sum使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类xarray.DataArray
的用法示例。
在下文中一共展示了DataArray.sum方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: showIncrVER
# 需要导入模块: from xarray import DataArray [as 别名]
# 或者: from xarray.DataArray import sum [as 别名]
def showIncrVER(tTC: np.ndarray, tReqInd: int, tctime: np.ndarray,
ver: xarray.DataArray, tver: xarray.DataArray,
titxt: str, makePlots: List[str]):
saveplot = False
z = ver.alt_km
lamb = ver.wavelength
# if 'spectra1d' in makePlots:
# b = np.trapz(ver, z, axis=1) # integrate along z, looking up magnetic zenith
# plotspectra(b, lamb)
if 'eigtime' in makePlots:
fg = figure(figsize=(11, 8), dpi=100, tight_layout=True)
ax = fg.gca()
pcm = ax.pcolormesh(tTC, z, tver.sum(axis=0), # sum over wavelength
edgecolors='none', cmap=None, norm=None,
vmin=0, vmax=1e3)
ax.axvline(tTC[tReqInd], color='white', linestyle='--', label='Req. Time')
ax.axvline(tctime['tstartPrecip'], color='red', linestyle='--', label='Precip. Start')
ax.axvline(tctime['tendPrecip'], color='red', linestyle='--', label='Precip. End')
titlemean = titxt + (r'\n VER/flux: $\lambda \in$' +
str(lamb) + ' [nm]' +
'\n geodetic lat:' + str(tctime['latgeo_ini'])
+ ' lon:' + str(tctime['longeo_ini']) +
' date: ' + tctime['dayofsim'].strftime('%Y-%m-%d'))
# make room for long title
fg.subplots_adjust(top=0.8)
ax.set_title(titlemean, fontsize=9)
ax.yaxis.set_major_locator(MultipleLocator(100))
ax.yaxis.set_minor_locator(MultipleLocator(20))
# ax.xaxis.set_major_locator(MinuteLocator(interval=10))
ax.xaxis.set_major_locator(MinuteLocator(interval=1))
ax.xaxis.set_minor_locator(SecondLocator(interval=10))
ax.xaxis.set_major_formatter(DateFormatter('%H:%M:%S'))
ax.tick_params(axis='both', which='both', direction='out', labelsize=12)
ax.autoscale(True, tight=True)
cbar = fg.colorbar(pcm)
cbar.set_label('VER/flux', labelpad=0)
ax.set_xlabel('Time [UTC]')
ax.set_ylabel('altitude [km]')
if saveplot:
sfn = ''.join(e for e in titxt if e.isalnum() or e == '.') # remove special characters
fg.savefig('out/VER' + sfn + '.png', dpi=150, bbox_inches='tight')
close(fg)
if 'eigtime1d' in makePlots:
fg = figure(figsize=(11, 8), dpi=100)
ax = fg.gca()
# fg.subplots_adjust(top=0.85)
thistitle = titxt + ': {:d} emission lines\n VER/flux: geodetic lat: {} lon: {} {}'.format(
ver.shape[0], tctime['latgeo_ini'], tctime['longeo_ini'], tTC[tReqInd])
ax.set_title(thistitle, fontsize=12)
ax.set_xlabel('VER/flux')
ax.set_ylabel('altitude [km]')
for ifg, clamb in enumerate(lamb):
ax.semilogx(ver.iloc[ifg, :], z, label=str(clamb))
ax.yaxis.set_major_locator(MultipleLocator(100))
ax.yaxis.set_minor_locator(MultipleLocator(20))
ax.grid(True)
if ver.shape[0] < 20:
ax.legend(loc='upper center', bbox_to_anchor=(1.05, .95),
ncol=1, fancybox=True, shadow=True, fontsize=9)
ax.tick_params(axis='both', which='both', direction='in', labelsize=12)
ax.set_xlim(1e-9, 1e3)
ax.set_ylim((z[0], z[-1]))
if saveplot:
sfn = ''.join(e for e in titxt if e.isalnum()) # remove special characters
fg.savefig('out/VER' + sfn + '.png', dpi=150, bbox_inches='tight')
close(fg)