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