def y(self):
"""y軸.
Returns
-------
np.ndarray
y軸
"""
return np.arange(*utils.slice2tuple(self.yslice))
def z(self):
"""z軸.
Returns
-------
np.ndarray
z軸
"""
return np.arange(*utils.slice2tuple(self.zslice))
def x(self):
"""x軸.
Returns
-------
np.ndarray
x軸
"""
return np.arange(*utils.slice2tuple(self.xslice))
def axis(self, ax):
index = self.slice_axes[ax]
axis_slice = self.slices[index]
return np.array(*utils.slice2tuple(axis_slice))
def plot(self,
axes='auto',
show=False,
use_si=False,
offsets=None,
**kwargs):
"""2次元データをプロットする.
Parameters
----------
axes : str, optional
プロットする軸('xy', 'zx', etc), by default 'auto'
show : bool
プロットを表示する場合True(ファイルに保存する場合は非表示), by default False
use_si : bool
SI単位系を用いる場合True(そうでない場合EMSES単位系を用いる), by default False
offsets : (float or str, float or str, float or str)
プロットのx,y,z軸のオフセット('left': 最初を0, 'center': 中心を0, 'right': 最後尾を0, float: 値だけずらす), by default None
mesh : (numpy.ndarray, numpy.ndarray), optional
メッシュ, by default None
savefilename : str, optional
保存するファイル名(Noneの場合保存しない), by default None
cmap : matplotlib.Colormap or str or None, optional
カラーマップ, by default cm.coolwarm
vmin : float, optional
最小値, by default None
vmax : float, optional
最大値, by default None
figsize : (float, float), optional
図のサイズ, by default None
xlabel : str, optional
x軸のラベル, by default None
ylabel : str, optional
y軸のラベル, by default None
title : str, optional
タイトル, by default None
interpolation : str, optional
用いる補間方法, by default 'bilinear'
dpi : int, optional
解像度(figsizeが指定された場合は無視される), by default 10
Returns
-------
AxesImage or None
プロットしたimageデータ(保存またはshowした場合None)
Raises
------
Exception
プロットする軸のパラメータが間違っている場合の例外
Exception
プロットする軸がデータにない場合の例外
Exception
データの次元が2でない場合の例外
"""
if axes == 'auto':
axes = ''.join(sorted(self.objs[0].use_axes))
if not re.match(r'x[yzt]|y[xzt]|z[xyt]|t[xyz]', axes):
raise Exception(
'Error: axes "{axes}" cannot be used with Data2d'.format(
axes=axes))
if axes[0] not in self.objs[0].use_axes or axes[1] not in self.objs[
0].use_axes:
raise Exception(
'Error: axes "{axes}" cannot be used because {axes}-axis does not exist in this data.'
.format(axes=axes))
if len(self.objs[0].shape) != 2:
raise Exception(
'Error: axes "{axes}" cannot be used because data is not 2dim shape.'
.format(axes=axes))
# x: 3, y: 2, z:1 t:0
axis1 = self.objs[0].slice_axes[self.objs[0].use_axes.index(axes[0])]
axis2 = self.objs[0].slice_axes[self.objs[0].use_axes.index(axes[1])]
x = np.arange(*utils.slice2tuple(self.objs[0].slices[axis1]))
y = np.arange(*utils.slice2tuple(self.objs[0].slices[axis2]))
if use_si:
xunit = self.objs[0].axisunits[axis1]
yunit = self.objs[0].axisunits[axis2]
valunit = self.objs[0].valunit
x = xunit.reverse(x)
y = yunit.reverse(y)
_xlabel = '{} [{}]'.format(axes[0], xunit.unit)
_ylabel = '{} [{}]'.format(axes[1], yunit.unit)
_title = '{} [{}]'.format(self.objs[0].name, valunit.unit)
x_data = self.x_data.val_si
y_data = self.y_data.val_si
else:
_xlabel = axes[0]
_ylabel = axes[1]
_title = self.objs[0].name
x_data = self.x_data
y_data = self.y_data
def _offseted(line, offset):
if offset == 'left':
line -= line[0]
elif offset == 'center':
line -= line[len(line) // 2]
elif offset == 'right':
line -= line[-1]
else:
line += offset
return line
if offsets is not None:
x = _offseted(x, offsets[0])
y = _offseted(y, offsets[1])
kwargs['xlabel'] = kwargs.get('xlabel', None) or _xlabel
kwargs['ylabel'] = kwargs.get('ylabel', None) or _ylabel
kwargs['title'] = kwargs.get('title', None) or _title
mesh = np.meshgrid(x, y)
img = emplt.plot_2d_vector(x_data, y_data, mesh=mesh, **kwargs)
if show:
plt.show()
return None
else:
return img
def plot(self, show=False, use_si=True, offsets=None, **kwargs):
"""1次元データをプロットする.
Parameters
----------
show : bool
プロットを表示する場合True(ファイルに保存する場合は非表示), by default False
use_si : bool
SI単位系を用いる場合True(そうでない場合EMSES単位系を用いる), by default True
offsets : (float or str, float or str)
プロットのx,y軸のオフセット('left': 最初を0, 'center': 中心を0, 'right': 最後尾を0, float: 値だけずらす), by default None
savefilename : str, optional
保存するファイル名, by default None
vmin : float, optional
最小値, by default None
vmax : float, optional
最大値, by default None
figsize : (float, float), optional
図のサイズ, by default None
xlabel : str, optional
横軸のラベル, by default None
ylabel : str, optional
縦軸のラベル, by default None
label : str, optional
ラベル, by default None
title : str, optional
タイトル, by default None
Returns
-------
Line2D or None
プロットデータを表す線オブジェクト(保存または show した場合None)
Raises
------
Exception
データの次元が1でない場合の例外
"""
if len(self.shape) != 1:
raise Exception(
'Error: cannot plot because data is not 1dim shape.')
axis = self.slice_axes[0]
x = np.arange(*utils.slice2tuple(self.slices[axis]))
y = self
# "EMSES Unit" to "Physical Unit"
if use_si:
xunit = self.axisunits[axis]
x = xunit.reverse(x)
y = self.valunit.reverse(y)
_xlabel = '{} [{}]'.format(self.use_axes[0], xunit.unit)
_ylabel = '{} [{}]'.format(self.name, self.valunit.unit)
else:
_xlabel = self.use_axes[0]
_ylabel = self.name
def _offseted(line, offset):
if offset == 'left':
line -= line[0]
elif offset == 'center':
line -= line[len(line) // 2]
elif offset == 'right':
line -= line[-1]
else:
line += offset
return line
if offsets is not None:
x = _offseted(x, offsets[0])
y = _offseted(y, offsets[1])
kwargs['xlabel'] = kwargs.get('xlabel', None) or _xlabel
kwargs['ylabel'] = kwargs.get('ylabel', None) or _ylabel
line = emplt.plot_line(y, x=x, **kwargs)
if show:
plt.show()
return None
else:
return line
def plot(self,
axes='auto',
show=False,
use_si=True,
offsets=None,
mode='cm',
**kwargs):
"""2次元データをプロットする.
Parameters
----------
axes : str, optional
プロットする軸('xy', 'zx', etc), by default 'auto'
show : bool
プロットを表示する場合True(ファイルに保存する場合は非表示), by default False
use_si : bool
SI単位系を用いる場合True(そうでない場合EMSES単位系を用いる), by default True
offsets : (float or str, float or str, float or str)
プロットのx,y,z軸のオフセット('left': 最初を0, 'center': 中心を0, 'right': 最後尾を0, float: 値だけずらす), by default None
mode : str
プロットの種類('cm': カラーマップ, 'cont': 等高線プロット, 'surf': サーフェースプロット)
mesh : (numpy.ndarray, numpy.ndarray), optional
メッシュ, by default None
savefilename : str, optional
保存するファイル名(Noneの場合保存しない), by default None
cmap : matplotlib.Colormap or str or None, optional
カラーマップ, by default cm.coolwarm
vmin : float, optional
最小値, by default None
vmax : float, optional
最大値, by default None
figsize : (float, float), optional
図のサイズ, by default None
xlabel : str, optional
x軸のラベル, by default None
ylabel : str, optional
y軸のラベル, by default None
title : str, optional
タイトル, by default None
interpolation : str, optional
用いる補間方法, by default 'bilinear'
dpi : int, optional
解像度(figsizeが指定された場合は無視される), by default 10
Returns
-------
AxesImage or None
プロットしたimageデータ(保存またはshowした場合None)
Raises
------
Exception
プロットする軸のパラメータが間違っている場合の例外
Exception
プロットする軸がデータにない場合の例外
Exception
データの次元が2でない場合の例外
"""
if axes == 'auto':
axes = ''.join(sorted(self.use_axes))
if not re.match(r'x[yzt]|y[xzt]|z[xyt]|t[xyz]', axes):
raise Exception(
'Error: axes "{axes}" cannot be used with Data2d'.format(
axes=axes))
if axes[0] not in self.use_axes or axes[1] not in self.use_axes:
raise Exception(
'Error: axes "{axes}" cannot be used because {axes}-axis does not exist in this data.'
.format(axes=axes))
if len(self.shape) != 2:
raise Exception(
'Error: axes "{axes}" cannot be used because data is not 2dim shape.'
.format(axes=axes))
# x: 3, y: 2, z:1 t:0
axis1 = self.slice_axes[self.use_axes.index(axes[0])]
axis2 = self.slice_axes[self.use_axes.index(axes[1])]
x = np.arange(*utils.slice2tuple(self.slices[axis1]))
y = np.arange(*utils.slice2tuple(self.slices[axis2]))
z = self if axis1 > axis2 else self.T # 'xz'等の場合は転置
if use_si:
xunit = self.axisunits[axis1]
yunit = self.axisunits[axis2]
x = xunit.reverse(x)
y = yunit.reverse(y)
z = self.valunit.reverse(z)
_xlabel = '{} [{}]'.format(axes[0], xunit.unit)
_ylabel = '{} [{}]'.format(axes[1], yunit.unit)
_title = '{} [{}]'.format(self.name, self.valunit.unit)
else:
_xlabel = axes[0]
_ylabel = axes[1]
_title = self.name
def _offseted(line, offset):
if offset == 'left':
line -= line.ravel()[0]
elif offset == 'center':
line -= line.ravel()[line.size // 2]
elif offset == 'right':
line -= line.ravel()[-1]
else:
line += offset
return line
kwargs['xlabel'] = kwargs.get('xlabel', None) or _xlabel
kwargs['ylabel'] = kwargs.get('ylabel', None) or _ylabel
kwargs['title'] = kwargs.get('title', None) or _title
if mode == 'surf':
mesh = np.meshgrid(x, y)
kwargs['zlabel'] = kwargs.get('zlabel', None) or _title
val = z
if 'x' not in self.use_axes:
y, z = mesh
x = self.x_si[0] if use_si else self.x[0]
x = np.zeros_like(mesh[0]) + x
elif 'y' not in self.use_axes:
x, z = mesh
y = self.y_si[0] if use_si else self.y[0]
y = np.zeros_like(mesh[0]) + y
elif 'z' not in self.use_axes:
x, y = mesh
z = self.z_si[0] if use_si else self.z[0]
z = np.zeros_like(mesh[0]) + z
if offsets is not None:
x = _offseted(x, offsets[0])
y = _offseted(y, offsets[1])
z = _offseted(z, offsets[2])
val = _offseted(val, offsets[3])
imgs = [emplt.plot_surface(x, y, z, val, **kwargs)]
else:
if offsets is not None:
x = _offseted(x, offsets[0])
y = _offseted(y, offsets[1])
z = _offseted(z, offsets[2])
mesh = np.meshgrid(x, y)
imgs = []
if 'cm' in mode and 'cont' in mode:
savefilename = kwargs.get('savefilename', None)
kwargs['savefilename'] = None
img = emplt.plot_2dmap(z, mesh=mesh, **kwargs)
kwargs['savefilename'] = savefilename
img2 = emplt.plot_2d_contour(z, mesh=mesh, **kwargs)
imgs = [img, img2]
elif 'cm' in mode:
img = emplt.plot_2dmap(z, mesh=mesh, **kwargs)
imgs.append(img)
elif 'cont' in mode:
img = emplt.plot_2d_contour(z, mesh=mesh, **kwargs)
imgs.append(img)
if show:
plt.show()
return None
else:
return imgs[0] if len(imgs) == 1 else imgs