DOC: simplified examples in DataFrame.assign docstring

Author: aeltanawy

pandas/core/frame.py

@@ -3354,23 +3354,25 @@ def assign(self, **kwargs):
 
  Examples
  --------
- >>> df = pd.DataFrame({'A': range(1, 3),
- ... 'B': np.arange(-1.0, 2.0, 1.5)})
+ >>> df = pd.DataFrame([('or', 17.),('ca', 25)],
+ index=['portland', 'berkeley'],
+ columns=['state', 'temp_c'])
 
  Where the value is a callable, evaluated on `df`:
 
- >>> df.assign(ln_A = lambda x: np.log(x.A))
- A B ln_A
- 0 1 -1.0 0.000000
- 1 2 0.5 0.693147
+ >>> df.assign(state=lambda x: pd.Categorical(x.state),
+ temp_f=lambda x: x.temp_c * 9 / 5 + 32)
+ state temp_c
+ portland or 17.0
+ berkeley ca 25.0
 
  Where the value already exists and is inserted:
 
- >>> newcol = np.log(df['A'])
- >>> df.assign(ln_A=newcol)
- A B ln_A
-  0 1 -1.0 0.000000
-  1 2 0.5 0.693147
+ >>> newcol = df['temp_c'] * 9 / 5 + 32
+ >>> df.assign(temp_f=newcol)
+  state temp_c temp_f
+ portland or 17.0  62.6
+ berkeley ca 25.0 77.0
 
  Where the keyword arguments depend on each other