python - 使用 Pandas 数据框时，如果不存在，如何添加列？

Question

我是使用 pandas 的新手，正在编写一个脚本，我在其中读取数据框，然后对某些列进行一些计算。

有时我会有一个名为“Met”的专栏:

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

其他时候我会有:

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

我需要对“Met”列进行一些计算，所以如果它不存在，我需要使用以下方法进行计算:

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

有没有办法检查“Met”列是否存在于数据框中，如果没有则添加它？

Answer 1

你可以这样检查:

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