|
5 | 5 |
|
6 | 6 | import astropy.units as u |
7 | 7 | import numpy as np |
8 | | -import pandas as pd |
9 | 8 | from astropy.time import Time |
10 | | -from astropy.table import Table |
| 9 | +from astropy.table import Table, vstack |
11 | 10 | import astropy.units as u |
12 | 11 |
|
13 | 12 | AST_DTYPE = np.dtype( |
@@ -558,30 +557,19 @@ def entire_db(): |
558 | 557 | ast_database = asteroid_db() |
559 | 558 | com_database = comet_db() |
560 | 559 |
|
561 | | - ast_database = pd.DataFrame( |
| 560 | + ast_database = Table( |
562 | 561 | ast_database[ |
563 | 562 | list(ast_database.dtype.names[:17]) |
564 | 563 | + list(ast_database.dtype.names[-4:-3]) |
565 | 564 | + list(ast_database.dtype.names[-2:]) |
566 | 565 | ] |
567 | 566 | ) |
568 | | - ast_database.rename( |
569 | | - columns={"ASTNAM": "NAME", "NO": "NUMBER", "CALEPO": "CALEPOCH"}, |
570 | | - inplace=True |
571 | | - ) |
572 | | - com_database = pd.DataFrame( |
| 567 | + com_database = Table( |
573 | 568 | com_database[ |
574 | 569 | list(com_database.dtype.names[:17]) |
575 | 570 | + list(com_database.dtype.names[-4:-3]) |
576 | 571 | + list(com_database.dtype.names[-2:]) |
577 | 572 | ] |
578 | 573 | ) |
579 | | - com_database.rename( |
580 | | - columns={"COMNAM": "NAME", "NO": "NUMBER", "CALEPO": "CALEPOCH"}, |
581 | | - inplace=True |
582 | | - ) |
583 | | - df = ast_database.append(com_database, ignore_index=True) |
584 | | - df[["NAME", "DESIG", "IREF"]] = df[["NAME", "DESIG", "IREF"]].apply( |
585 | | - lambda x: x.str.strip().str.decode("utf-8") |
586 | | - ) |
| 574 | + df = vstack(ast_database, com_database) |
587 | 575 | return df |
0 commit comments