Dataframe中的数据筛选

大家好!

我现在有名为stock的Dataframe,如下图:

img

我现在希望这个Dataframe中只保留ts_code列的值是以'SH'或'SZ'结尾的行的数据。
请问用什么代码可以实现,谢谢!

可以通过pandas条件筛选完成,以下为参考代码

filtered_stock = stock[stock['ts_code'].str.endswith(('SH', 'SZ'))]

其中条件部分可以根据情况调整和设置


stock=stock[stock['ts_code'].str[-2:]=="SH"]