20行Python代码教会“钢铁直男”分清颜色