[ARROW-18189] [Python] Table.drop should support passing a single column

• Type: Improvement
• Status: Open
• Priority: Major
• Resolution: Unresolved
• Component/s: Python
• Labels: good-first-issue
• External issue URL: https://github.com/apache/arrow/issues/33377
• Assignee: Unassigned
• Reporter: Alessandro Molina
• Created: 28/Oct/22 13:48
• Updated: 11/Jan/23 11:58

Description

----> 2 data = dataset.drop("Churn") 

/usr/local/lib/python3.7/dist-packages/pyarrow/table.pxi in pyarrow.lib.Table.drop()
    KeyError: "Column 'C' not found"

Also, for consistency, it would probably be good to have a Table.drop_column alias, as all the other methods are named Table.add_column, Table.append_column and Table.set_column