Add support for column-or-listofcolumns style value for key_field parameter, like that of many functions in standard library