You can just read the input, convert it to an integer and use it directly as the column name.
Here is a simple example:

df = pd.DataFrame({1: [10, 20, 30],2: [40, 50, 60]})

x = int(input("enter your value here: ")) 

A = df[x]

print(A)