如何在Pandas中从dataframepython3获取特定列的用户输入

Question

我正在使用PANDAS Dataframe从CSV文件中获取数据，并选择所需的列数据，该数据运行良好。但是，在hostData数据帧中，我特别寻找(data['Safe']=='KDS-PDC-DEFAULT-UNIX-ROOT')部分，因为我在文档字符串中的CyberArk Safes下面提到了多个保险箱，因此我希望在用户输入时，将安全名称手动放在代码上。在

当请求用户输入给出安全名称时，它是否可以显示文档字符串中的当前安全名称。在

#!/grid/common/pkgs/python/v3.6.1/bin/python3
from __future__ import print_function
from signal import signal, SIGPIPE, SIG_DFL
signal(SIGPIPE,SIG_DFL)
import csv
import pandas as pd

##### Python pandas, widen output display to see more columns. ####
pd.set_option('display.height', None)
pd.set_option('display.max_rows', None)
pd.set_option('display.max_columns', None)
pd.set_option('display.width', None)
pd.set_option('expand_frame_repr', True)

############# CyberArk Safes ################
'''  
KDS-KDC-DEFAULT-UNIX-ROOT
KDS-PDC-DEFAULT-UNIX-ROOT
KDS-CDC-DEFAULT-UNIX-ROOT
'''

################# PANDAS Extraction ###########
data = pd.read_csv('/home/karn/plura/Test/Python_Panda/InventoryReports06.csv', usecols=['Platform ID', 'Safe', 'Target system address', 'Failure reason'])
hostData = data[(data['Safe']=='DC') | (data['Safe']=='KDS-PDC-DEFAULT-UNIX-ROOT')][['Safe', 'Target system address']]
hostData.reset_index(level=0, drop=True)

预期产量：

^{pr2}$

任何帮助或建议将不胜感激。在

Answer 1

1. 要显示候选对象，请创建一个函数（例如print_candidates()）以显示docstring。在
2. 要获取用户输入，请使用input()并将其存储到SafeName，然后使用它创建一个掩码来过滤数据帧。在

因此，代码的后面部分如下所示：

def print_candidates():
    '''  
    KDS-KDC-DEFAULT-UNIX-ROOT
    KDS-PDC-DEFAULT-UNIX-ROOT
    KDS-CDC-DEFAULT-UNIX-ROOT
    '''
    print("The Current SafeName Available are:")
    print(print_candidates.__doc__)

print_candidates()

################# PANDAS Extraction ###########
data = pd.read_csv('/home/karn/plura/Test/Python_Panda/InventoryReports06.csv', usecols=['Platform ID', 'Safe', 'Target system address', 'Failure reason'])
SafeName = str(input("Please Enter the SafeName:"))
hostData = data.loc[(data['Safe']=='DC') | (data['Safe']==SafeName)][['Safe', 'Target system address']]
hostData = hostData.reset_index(drop=True)