Delete data frame column if column name ends with some string, Python 3.6

user_2717063 0
Viewed 11718

I have a dataframe with below columns:

SectorName', 'Sector', 'ItemName', 'Item', 'Counterpart SectorName', 'Counterpart Sector', 'Stocks and TransactionsName', 'Stocks and Transactions', 'Units', 'Scale', 'Frequency', 'Date', 'Value'

How to delete column from df where column name ends with Name.

pandas python-3.x python
1 Answers

You can filter by inverting (~) boolean mask for columns which not need delete with loc and str.endswith, also working str.contains with $ for match end of string:

cols = ['SectorName', 'Name Sector', 'ItemName', 'Item', 'Counterpart SectorName']
df = pd.DataFrame([range(5)], columns = cols)
print (df)
   SectorName  Name Sector  ItemName  Item  Counterpart SectorName
0           0            1         2     3                       4

print (~df.columns.str.endswith('Name'))
[False  True False  True False]

df1 = df.loc[:, ~df.columns.str.endswith('Name')]

df1 = df.loc[:, ~df.columns.str.contains('Name$')]

Or filter columns names first:

print (df.columns[~df.columns.str.endswith('Name')])
Index(['Sector', 'Item'], dtype='object')

df1 = df[df.columns[~df.columns.str.endswith('Name')]]

print (df1)
   Name Sector  Item
0            1     3
user_29010020
Related