Extract specific values from a column through pandas

Question

I have a vcf file which contains multiple columns: Columns 1

INFO Columns contain multiple values in dictionary: INFO 2

I just want to extract "AF" values from INFO Column and make plot against chromosome. How can I do this?

Answer 1

You can use apply function to extract values from pandas column

import pandas as pd


# Sample data
df = pd.DataFrame({"INFO": [{"AF": [0.133], "AC": [35], "AN": 264}, {"AF": [0.007353], "AC": [2], "AN": 272}, {"AF": [0.026], "AC": [7], "AN": 272}], "POS": [50041, 50047, 50072]})

df["AF"] = df["INFO"].apply(lambda x: x["AF"])
print(df)