2014-09-17 13 views
5

Ich bin neu in der Verwendung von Pandas und schreibe ein Skript, wo ich in einem Datenframe lesen und dann einige Berechnungen auf einige der Spalten tun.Pandas: Spalte hinzufügen, wenn nicht existiert

Manchmal habe ich die Spalte "Met" genannt:

df = pd.read_csv(File, sep='\t', compression='gzip', header=0, names=["Chrom", "Site", "coverage", "Met"]) 

Othertimes werde ich:

df = pd.read_csv(File, sep='\t', compression='gzip', header=0, names=["Chrom", "Site", "coverage", "freqC"]) 

Ich brauche eine Berechnung mit der "Met" Spalte zu tun, wenn es isn ‚t vorhanden ich muss es berechnen mit:

df['Met'] = df['freqC'] * df['coverage'] 

ist es eine Möglichkeit, wenn die‚Met‘zu überprüfen Spalte pres ent in den Datenrahmen, und wenn nicht hinzufügen?

Antwort

15

Sie überprüfen Sie es wie folgt aus:

if 'Met' not in df: 
    df['Met'] = df['freqC'] * df['coverage'] 
Verwandte Themen