Density of states collections

The dosdata and doscollection modules contained in ase.spectrum form a new framework, intended to replace the DOS implementations in ase.dft.dos and ase.dft.pdos.

This module provides DOSCollection and GridDOSCollection objects that contain a set of related DOS data objects. This would typically represent partitioned contributions to a density-of-states, such as atomic contributions to a vibrational spectrum or orbital contributions to an electronic spectrum.

More details

class ase.spectrum.doscollection.DOSCollection(dos_series: Iterable[DOSData])[source]

Base class for a collection of DOSData objects

classmethod from_data(energies: Sequence[float] | ndarray, weights: Sequence[Sequence[float] | ndarray], info: Sequence[Dict[str, str]] = None) DOSCollection[source]

Create a DOSCollection from data sharing a common set of energies

This is a convenience method to be used when all the DOS data in the collection has a common energy axis. There is no performance advantage in using this method for the generic DOSCollection, but for GridDOSCollection it is more efficient.

Parameters:
  • energy – common set of energy values for input data

  • weights – array of DOS weights with rows corresponding to different datasets

  • info – sequence of info dicts corresponding to weights rows.

Returns:

Collection of DOS data (in RawDOSData format)

plot(npts: int = 1000, xmin: float = None, xmax: float = None, width: float = 0.1, smearing: str = 'Gauss', ax: Axes = None, show: bool = False, filename: str = None, mplargs: dict = None) Axes[source]

Simple plot of collected DOS data, resampled onto a grid

If the special key ‘label’ is present in self.info, this will be set as the label for the plotted line (unless overruled in mplargs). The label is only seen if a legend is added to the plot (i.e. by calling \(ax.legend()\)).

Parameters:
  • npts – output data range, as passed to self.sample_grid

  • xmin – output data range, as passed to self.sample_grid

  • xmax – output data range, as passed to self.sample_grid

  • width – Width of broadening kernel, passed to self.sample_grid()

  • smearing – selection of broadening kernel for self.sample_grid()

  • ax – existing Matplotlib axes object. If not provided, a new figure with one set of axes will be created using Pyplot

  • show – show the figure on-screen

  • filename – if a path is given, save the figure to this file

  • mplargs – additional arguments to pass to matplotlib plot command (e.g. {‘linewidth’: 2} for a thicker line).

Returns:

Plotting axes. If “ax” was set, this is the same object.

sample_grid(npts: int, xmin: float = None, xmax: float = None, padding: float = 3, width: float = 0.1, smearing: str = 'Gauss') GridDOSCollection[source]

Sample the DOS data on an evenly-spaced energy grid

Parameters:
  • npts – Number of sampled points

  • xmin – Minimum sampled energy value; if unspecified, a default is chosen

  • xmax – Maximum sampled energy value; if unspecified, a default is chosen

  • padding – If xmin/xmax is unspecified, default value will be padded by padding * width to avoid cutting off peaks.

  • width – Width of broadening kernel, passed to self.sample_grid()

  • smearing – selection of broadening kernel, for self.sample_grid()

Returns:

(energy values, sampled DOS)

select(**info_selection: str) DOSCollection[source]

Narrow DOSCollection to items with specified info

For example, if

dc = DOSCollection([DOSData(x1, y1, info={'a': '1', 'b': '1'}),
                    DOSData(x2, y2, info={'a': '2', 'b': '1'})])

then

dc.select(b='1')

will return an identical object to dc, while

dc.select(a='1')

will return a DOSCollection with only the first item and

dc.select(a='2', b='1')

will return a DOSCollection with only the second item.

select_not(**info_selection: str) DOSCollection[source]

Narrow DOSCollection to items without specified info

For example, if

dc = DOSCollection([DOSData(x1, y1, info={'a': '1', 'b': '1'}),
                    DOSData(x2, y2, info={'a': '2', 'b': '1'})])

then

dc.select_not(b='2')

will return an identical object to dc, while

dc.select_not(a='2')

will return a DOSCollection with only the first item and

dc.select_not(a='1', b='1')

will return a DOSCollection with only the second item.

sum_all() DOSData[source]

Sum all the DOSData contained in this Collection

sum_by(*info_keys: str) DOSCollection[source]

Return a DOSCollection with some data summed by common attributes

For example, if

dc = DOSCollection([DOSData(x1, y1, info={'a': '1', 'b': '1'}),
                    DOSData(x2, y2, info={'a': '2', 'b': '1'}),
                    DOSData(x3, y3, info={'a': '2', 'b': '2'})])

then

dc.sum_by('b')

will return a collection equivalent to

DOSCollection([DOSData(x1, y1, info={'a': '1', 'b': '1'})
               + DOSData(x2, y2, info={'a': '2', 'b': '1'}),
               DOSData(x3, y3, info={'a': '2', 'b': '2'})])

where the resulting contained DOSData have info attributes of {‘b’: ‘1’} and {‘b’: ‘2’} respectively.

dc.sum_by(‘a’, ‘b’) on the other hand would return the full three-entry collection, as none of the entries have common ‘a’ and ‘b’ info.

total() DOSData[source]

Sum all the DOSData in this Collection and label it as ‘Total’

class ase.spectrum.doscollection.GridDOSCollection(dos_series: Iterable[GridDOSData], energies: Sequence[float] | ndarray | None = None)[source]
classmethod from_data(energies: Sequence[float] | ndarray, weights: Sequence[Sequence[float] | ndarray], info: Sequence[Dict[str, str]] = None) GridDOSCollection[source]

Create a GridDOSCollection from data with a common set of energies

This convenience method may also be more efficient as it limits redundant copying/checking of the data.

Parameters:
  • energies – common set of energy values for input data

  • weights – array of DOS weights with rows corresponding to different datasets

  • info – sequence of info dicts corresponding to weights rows.

Returns:

Collection of DOS data (in RawDOSData format)

plot(npts: int = 0, xmin: float = None, xmax: float = None, width: float = None, smearing: str = 'Gauss', ax: Axes = None, show: bool = False, filename: str = None, mplargs: dict = None) Axes[source]

Simple plot of collected DOS data, resampled onto a grid

If the special key ‘label’ is present in self.info, this will be set as the label for the plotted line (unless overruled in mplargs). The label is only seen if a legend is added to the plot (i.e. by calling \(ax.legend()\)).

Parameters:
  • npts – Number of points in resampled x-axis. If set to zero (default), no resampling is performed and the stored data is plotted directly.

  • xmin – output data range; this limits the resampling range as well as the plotting output

  • xmax – output data range; this limits the resampling range as well as the plotting output

  • width – Width of broadening kernel, passed to self.sample()

  • smearing – selection of broadening kernel, passed to self.sample()

  • ax – existing Matplotlib axes object. If not provided, a new figure with one set of axes will be created using Pyplot

  • show – show the figure on-screen

  • filename – if a path is given, save the figure to this file

  • mplargs – additional arguments to pass to matplotlib plot command (e.g. {‘linewidth’: 2} for a thicker line).

Returns:

Plotting axes. If “ax” was set, this is the same object.

select(**info_selection: str) DOSCollection[source]

Narrow GridDOSCollection to items with specified info

For example, if

dc = GridDOSCollection([GridDOSData(x, y1,
                                    info={'a': '1', 'b': '1'}),
                        GridDOSData(x, y2,
                                    info={'a': '2', 'b': '1'})])

then

dc.select(b='1')

will return an identical object to dc, while

dc.select(a='1')

will return a DOSCollection with only the first item and

dc.select(a='2', b='1')

will return a DOSCollection with only the second item.

select_not(**info_selection: str) DOSCollection[source]

Narrow GridDOSCollection to items without specified info

For example, if

dc = GridDOSCollection([GridDOSData(x, y1,
                                    info={'a': '1', 'b': '1'}),
                        GridDOSData(x, y2,
                                    info={'a': '2', 'b': '1'})])

then

dc.select_not(b='2')

will return an identical object to dc, while

dc.select_not(a='2')

will return a DOSCollection with only the first item and

dc.select_not(a='1', b='1')

will return a DOSCollection with only the second item.

class ase.spectrum.doscollection.RawDOSCollection(dos_series: Iterable[RawDOSData])[source]