"""
select - Select data table subsets based on multiple spatial criteria.
"""
from typing import Literal
import numpy as np
import pandas as pd
from pygmt.clib import Session
from pygmt.helpers import (
build_arg_list,
fmt_docstring,
kwargs_to_strings,
use_alias,
validate_output_table_type,
)
__doctest_skip__ = ["select"]
[docs]
@fmt_docstring
@use_alias(
A="area_thresh",
C="dist2pt",
D="resolution",
F="polygon",
G="gridmask",
I="reverse",
J="projection",
L="dist2line",
N="mask",
R="region",
V="verbose",
Z="z_subregion",
b="binary",
d="nodata",
e="find",
f="coltypes",
g="gap",
h="header",
i="incols",
o="outcols",
s="skiprows",
w="wrap",
)
@kwargs_to_strings(M="sequence", R="sequence", i="sequence_comma", o="sequence_comma")
def select(
data=None,
output_type: Literal["pandas", "numpy", "file"] = "pandas",
outfile: str | None = None,
**kwargs,
) -> pd.DataFrame | np.ndarray | None:
r"""
Select data table subsets based on multiple spatial criteria.
This is a filter that reads (x, y) or (longitude, latitude) positions from
the first 2 columns of ``data`` and uses a combination of 1-7 criteria to
pass or reject the records. Records can be selected based on whether or not
they:
1. are inside a rectangular region (``region`` [and ``projection``])
2. are within *dist* km of any point in *pointfile* (``dist2pt``)
3. are within *dist* km of any line in *linefile* (``dist2line``)
4. are inside one of the polygons in *polygonfile* (``polygon``)
5. are inside geographical features (based on coastlines)
6. have z-values within a given range
7. are inside bins of a grid mask whose nodes are non-zero
The sense of the tests can be reversed for each of these 7 criteria by
using the ``reverse`` parameter.
Full option list at :gmt-docs:`gmtselect.html`
{aliases}
Parameters
----------
data : str, {table-like}
Pass in either a file name to an ASCII data table, a 2-D
{table-classes}.
{output_type}
{outfile}
{area_thresh}
dist2pt : str
*pointfile*\|\ *lon*/*lat*\ **+d**\ *dist*.
Pass all records whose locations are within *dist* of any of the
points in the ASCII file *pointfile*. If *dist* is zero, the 3rd
column of *pointfile* must have each point's individual radius of
influence. If you only have a single point, you can specify
*lon*/*lat* instead of *pointfile*. Distances are Cartesian and in
user units. Alternatively, if ``region`` and ``projection`` are used,
the geographic coordinates are projected to map coordinates (in
centimeters, inches, meters, or points, as determined by
:gmt-term:`PROJ_LENGTH_UNIT`) before Cartesian distances are compared
to *dist*.
dist2line : str
*linefile*\ **+d**\ *dist*\ [**+p**].
Pass all records whose locations are within *dist* of any of the line
segments in the ASCII :gmt-docs:`multiple-segment file
<reference/file-formats.html#optional-segment-header-records>`
*linefile*. If *dist* is zero, we will scan each sub-header in
*linefile* for an embedded **-D**\ *dist* setting that sets each
line's individual distance value. Distances are Cartesian and in
user units. Alternatively, if ``region`` and ``projection`` are used,
the geographic coordinates are projected to map coordinates (in
centimeters, inches, meters, or points, as determined by
:gmt-term:`PROJ_LENGTH_UNIT`) before Cartesian distances are
compared to *dist*. Append **+p** to ensure only points whose
orthogonal projections onto the nearest line-segment fall within
the segment's endpoints [Default considers points "beyond" the
line's endpoints].
polygon : str
*polygonfile*.
Pass all records whose locations are within one of the closed
polygons in the ASCII :gmt-docs:`multiple-segment file
<reference/file-formats.html#optional-segment-header-records>`
*polygonfile*. For spherical polygons (lon, lat), make sure no
consecutive points are separated by 180 degrees or more in longitude.
resolution : str
*resolution*\ [**+f**].
Ignored unless ``mask`` is set. Selects the resolution of the coastline
data set to use ((**f**)ull, (**h**)igh, (**i**)ntermediate, (**l**)ow,
or (**c**)rude). The resolution drops off by ~80% between data sets.
[Default is **l**]. Append (**+f**) to automatically select a lower
resolution should the one requested not be available [Default is abort
if not found]. Note that because the coastlines differ in details
it is not guaranteed that a point will remain inside [or outside] when
a different resolution is selected.
gridmask : str
Pass all locations that are inside the valid data area of the grid
*gridmask*. Nodes that are outside are either NaN or zero.
reverse : str
[**cflrsz**].
Reverse the sense of the test for each of the criteria specified:
- **c** select records NOT inside any point's circle of influence.
- **f** select records NOT inside any of the polygons.
- **g** will pass records inside the cells with z equal zero of the
grid mask in ``gridmask``.
- **l** select records NOT within the specified distance of any line.
- **r** select records NOT inside the specified rectangular region.
- **s** select records NOT considered inside as specified by ``mask``
(and ``area_thresh``, ``resolution``).
- **z** select records NOT within the range specified by
``z_subregion``.
{projection}
mask : str or list
Pass all records whose location is inside specified geographical
features. Specify if records should be skipped (s) or kept (k) using
1 of 2 formats:
- *wet/dry*.
- *ocean/land/lake/island/pond*.
[Default is s/k/s/k/s (i.e., s/k), which passes all points on dry
land].
{region}
{verbose}
z_subregion : str or list
*min*\ [/*max*]\ [**+a**]\ [**+c**\ *col*]\ [**+i**].
Pass all records whose 3rd column (*z*; *col* = 2) lies within the
given range or is NaN (use ``skiprows`` to skip NaN records). If *max*
is omitted then we test if *z* equals *min* instead. This means
equality within 5 ULPs (unit of least precision;
https://en.wikipedia.org/wiki/Unit_in_the_last_place). Input file must
have at least three columns. To indicate no limit on *min* or *max*,
specify a hyphen (-). If your 3rd column is absolute time then remember
to supply ``coltypes="2T"``. To specify another column, append
**+c**\ *col*, and to specify several tests pass a list of arguments
as you have columns to test.
**Note**: When more than one ``z_subregion`` argument is given then the
``reverse="z"`` cannot be used. In the case of multiple tests
you may use these modifiers as well: **+a** passes any record that
passes at least one of your *z* tests [Default is all tests must pass],
and **+i** reverses the tests to pass record with *z* value NOT in the
given range. Finally, if **+c** is not used then it is automatically
incremented for each new ``z_subregion`` argument, starting with 2.
{binary}
{nodata}
{find}
{coltypes}
{gap}
{header}
{incols}
{outcols}
{skiprows}
{wrap}
Returns
-------
ret
Return type depends on ``outfile`` and ``output_type``:
- ``None`` if ``outfile`` is set (output will be stored in file set by
``outfile``)
- :class:`pandas.DataFrame` or :class:`numpy.ndarray` if ``outfile`` is not set
(depends on ``output_type``)
Example
-------
>>> import pygmt
>>> # Load a table of ship observations of bathymetry off Baja California
>>> ship_data = pygmt.datasets.load_sample_data(name="bathymetry")
>>> # Only return the data points that lie within the region between
>>> # longitudes 246 and 247 and latitudes 20 and 21
>>> out = pygmt.select(data=ship_data, region=[246, 247, 20, 21])
"""
output_type = validate_output_table_type(output_type, outfile=outfile)
column_names = None
if output_type == "pandas" and isinstance(data, pd.DataFrame):
column_names = data.columns.to_list()
with Session() as lib:
with (
lib.virtualfile_in(check_kind="vector", data=data) as vintbl,
lib.virtualfile_out(kind="dataset", fname=outfile) as vouttbl,
):
lib.call_module(
module="select",
args=build_arg_list(kwargs, infile=vintbl, outfile=vouttbl),
)
return lib.virtualfile_to_dataset(
vfname=vouttbl,
output_type=output_type,
column_names=column_names,
)