def METHOD_NAME(df):
    msg_types = [col.name for col in df if col.dtype is pl.Int64]
    top_msgs = (
        df.select(msg_types)
        .sum()
        .transpose(include_header=True, header_name="msg", column_names=["count"])
        .top_k(args.k, by="count")
    )["msg"]
    other_msgs = set(msg_types) - set(top_msgs)
    return top_msgs, other_msgs