python - 是否有任何函数可以将数据帧列中的数字与一系列数字进行比较？

Question

我想检查一列是否包含从 1 到 9 的数字。但我不太明白该怎么做。

我的目标是验证用户在输入中输入的列是否包含从 1 到 9 的数字。如果该列包含从 1 到 9 的数字，则输出必须为: True 或 False 如果不包含从 1 到 9 的数字。

数据帧的内容如下:

   1  2  3  4  5  6  7  8  9
A  6  4  7  2  5  9  3  8  1
B  8  9  3  7  6  1  5  2  4
C  5  2  1  8  4  3  9  7  6
D  1  6  8  5  9  4  7  3  2
E  7  3  4  1  2  8  6  9  5
F  2  5  9  6  3  7  1  4  8
G  9  8  2  3  1  5  4  6  7
H  3  1  6  4  7  2  8  5  9
I  4  7  5  9  8  6  2  1  3

代码:

def validateCol(file, n_col):

    T = pd.read_fwf(file, header= None, names=['1','2','3','4','5','6','7','8','9'])
    T = T.rename(index={0:'A',1:'B',2:'C',3:'D',4:'E',5:'F',6:'G',7:'H',8:'I'})

    print(T)

    df = pd.DataFrame(T)
    result = df[n_col]
    print(result)

    #1. Validate if the content of n_col has the numbers from 1 to 9.
    #2. Return False or True depending on #1

validateCol(file=input('file name: '), n_col= input('n_col to validate: '))

预期输出:

>>> validateCol('file_name', 2)
The column contains the numbers from 1 to 9
True

Answer 1

你可以使用这样的东西:

check_list = [*range(1,10)] # [1, 2, 3, 4, 5, 6, 7, 8, 9]

def validateCol(file, n_col):

    T = pd.read_fwf(file, header= None, names=['1','2','3','4','5','6','7','8','9'])
    T = T.rename(index={0:'A',1:'B',2:'C',3:'D',4:'E',5:'F',6:'G',7:'H',8:'I'})

    print(T)

    df = pd.DataFrame(T)
    result = all(a in df[n_col].to_list() for a in check_list)
    if result:
        print("Column {} contains the numbers from 1 to 9".format(n_col))
    else:
        print("Column {} does not contain the numbers from 1 to 9".format(n_col))