把從的方法的值轉換成一個數據幀

Question

def number(): 
    for i in dataframe: 
     if i > 10: 
      print('large') 
     elif i <= 10: 
      print('small')     
number() 
small 
small 
large 
small 

是否有可以納入這些值的任何可能的方式，特別是那些從一種方法獲得，如上面的例子中爲DF諸如大熊貓數據幀？

Answer 1

與熊貓有關的主要問題是，在許多情況下，它期望矢量值運算符和函數。答案是用numpy

這似乎是做你想做的。請注意，其中大體上是您的函數的向量值等價值。

>>> import pandas as pd 
>>> import numpy as np 
>>> A = np.array([1,9,10,11,15]) 
>>> B = len(A)*[2] 
>>> df = pd.DataFrame({'A': A, 'B': B}) 
>>> df 
    A B 
0 1 2 
1 9 2 
2 10 2 
3 11 2 
4 15 2 
>>> df.assign(C=np.where(A>10,len(A)*['large'],len(A)*['small'])) 
    A B  C 
0 1 2 small 
1 9 2 small 
2 10 2 small 
3 11 2 large 
4 15 2 large