如何在没有“[]”的情况下获得值

Question

file_name = 'employee_handbook.csv'
cid = 2603704587

with open(file_name, newline = '') as handbook:
    emp_handbook = pd.DataFrame(pd.read_csv(file_name))
    row_number = emp_handbook[emp_handbook['CID'] == cid].index.values
    name = emp_handbook.iloc[row_number,[1]].values
print(name)

输出：

[['Jun Cheng']]

我怎样才能得到 'Jun Cheng' 而不是 [['Jun Cheng']]

Answer 1

你需要

print(yourlist[0][0][0])

其中包含 0 的括号对的数量是您想要索引以找到字符串的列表数量

例如

a = [['2222']]

print(a[0][0])

Answer 2

您可以使用以下代码：

file_name = 'employee_handbook.csv'
cid = 2603704587

emp_handbook = pd.read_csv(file_name)
name = emp_handbook.loc[lambda df: df.CID == cid, 'name'].iloc[0]

print(name)

也适用于您的情况：

• 无需使用 with 语句
• pd.read_csv 的返回值已经是 DataFrame，所以不需要添加 pd.DataFrame。