結果 : how to keep certain columns in python