pandas中loc函数选取的数据取Value相关操作及判断numpy数组是否为空

DataFrame用loc函数选择多列后组成的数据还是Dataframe，选取一列组成的数据是Series（备注：代码注释“#”后为运行结果）。

lat=sim_poi.loc[sim_poi.station == '上海火车站','gps_lat']
print(lat)
# 188    31.250757
# Name: gps_lat, dtype: float64

print("type:",type(lat))
# type: <class 'pandas.core.series.Series'>

取Series包含index和value，取其中的value：

lat_value=lat.values
print(type(lat_value))
# <class 'numpy.ndarray'>
print(lat_value)
#[ 31.25075675]
print(lat_value[0])
#31.2507567527

返回的是一维numpy数组。对于numpy数组判断是否为空，可以用数组中元素个数size是否为0:

if lat_value.size:
    print(lat_value.size)
#1

另外，对于loc取出的DataFrame，取出其中的Values得到一个多维数组，:

lat=sim_poi.loc[sim_poi.station != '上海火车站',['gps_lat','gps_lon']]
print(type(lat))
#<class 'pandas.core.frame.DataFrame'>
lat_value=lat.values
print(type(lat_value))
# <class 'numpy.ndarray'>
print(lat_value[0])
#[  31.28212776  121.54875723]
print(lat_value.shape)
#(302, 2)
print(lat_value.size)
#604