16 lines
508 B
Python
16 lines
508 B
Python
import adata
|
|
import sqlite3
|
|
|
|
conn = sqlite3.connect('big-a.db')
|
|
res_df = adata.stock.info.all_code()
|
|
|
|
no_exit = res_df['short_name'].map(lambda x: not x.endswith('退'))
|
|
have_date = res_df['list_date'].map(lambda x: not (not x))
|
|
no_st = res_df['short_name'].map(lambda x: 'ST' not in x)
|
|
no_pt = res_df['short_name'].map(lambda x: 'PT' not in x)
|
|
|
|
filtered = res_df[no_exit & no_st & no_pt & res_df['list_date'].notnull()]
|
|
filtered.to_sql('all_code', con=conn, if_exists='replace', index=False)
|
|
|
|
print('Done')
|