import pandas as pd
users = ["User1", "User2", "User3", "User4", "User5"]
items = ["Item A", "Item B", "Item C", "Item D", "Item E"]
datasets = [
[1,0,1,1,0],
[1,0,0,1,1],
[1,0,1,0,0],
[0,1,0,1,1],
[1,1,1,0,1],
]
df = pd.DataFrame(datasets,
columns=items,
index=users)
print(df)
"""
Item A Item B Item C Item D Item E
User1 1 0 1 1 0
User2 1 0 0 1 1
User3 1 0 1 0 0
User4 0 1 0 1 1
User5 1 1 1 0 1
"""
print(df.loc['User1',:])
"""
Item A 1
Item B 0
Item C 1
Item D 1
Item E 0
Name: User1, dtype: int64
"""
print(type(df.loc['User1',:]))
"""
<class 'pandas.core.series.Series'>
"""
print(df.loc[['User1'],:])
"""
Item A Item B Item C Item D Item E
User1 1 0 1 1 0
"""
print(type(df.loc[['User1'],:]))
"""
<class 'pandas.core.frame.DataFrame'>
"""
当使用dataframe.loc进行行或者列索引时,输出的是数据框还是序列要分清楚。