ENH: add ax keyword to pcolormesh