Skip to content

Instantly share code, notes, and snippets.

@nuada

nuada/drop_cols.py

Last active August 29, 2015 14:20
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save nuada/ba03b6cbdfcffa5dcb93 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save nuada/ba03b6cbdfcffa5dcb93 to your computer and use it in GitHub Desktop.
Download ZIP
Drop certain columns from TSV file using Pandas
Raw
drop_cols.py
#!/usr/bin/python
import pandas as pd
import sys
df = pd.read_csv(sys.argv[1], sep='\t', index_col=0)
for col in df.columns:
if col[3] == 'B':
df = df.drop(col, 1)
df.to_csv(sys.stdout, sep='\t')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment