Created
January 15, 2023 00:54
-
-
Save andycasey/4c73f17091a752b2113000bd7624adbb to your computer and use it in GitHub Desktop.
SDSS DR17 subset for Di Cook
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from astropy.table import Table | |
data = Table.read("/uufs/chpc.utah.edu/common/home/sdss50/dr17/apogee/spectro/aspcap/dr17/synspec_rev1/allStar-dr17-synspec_rev1.fits") | |
column_names = data.dtype.names | |
ignore = ["TIII_FE"] | |
available_elements = [ea for ea in column_names if f"{ea}_FLAG" in column_names and ea not in ignore] | |
keep = ( | |
(data["SNR"] >= 200) | |
& (data["TEFF"] >= 4000) | |
& (data["TEFF"] <= 7000) | |
& (data["ASPCAPFLAG"] == 0) | |
& (data["LOGG"] <= 4.4) | |
) | |
# remove columns with multiple dimensions | |
remove_column_names = ['FPARAM_GRID', 'CHI2_GRID', 'FPARAM', 'FPARAM_COV', 'PARAM', 'PARAM_COV', 'PARAMFLAG', 'FELEM', 'FELEM_ERR', 'X_H', 'X_H_ERR', 'X_M', 'X_M_ERR', 'ELEM_CHI2', 'ELEMFRAC', 'ELEMFLAG', 'X_H_SPEC', 'X_M_SPEC', 'VISIT_PK'] | |
remove_column_names.extend([ | |
"ASPCAP_ID", | |
"TARGFLAGS", | |
"ASPCAPFLAGS", | |
"PROGRAMNAME" | |
]) | |
for cn in remove_column_names: | |
if cn in data.dtype.names: | |
del data[cn] | |
data[keep].write("allStar-dr17-synspec_rev1-subset.csv", overwrite=True) | |
''' | |
import matplotlib.pyplot as plt | |
fig, ax = plt.subplots() | |
ax.scatter(data["TEFF"][keep], data["LOGG"][keep], c=data["FE_H"][keep], s=1) | |
ax.set_xlim(ax.get_xlim()[::-1]) | |
ax.set_ylim(ax.get_ylim()[::-1]) | |
fig.savefig("tmp.png", dpi=300) | |
''' |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment