Added substrings_in_string

pull/1/head
Carlos A. Iglesias 8 years ago
parent 1f7f6b68ae
commit 68f405d91d

@ -443,6 +443,12 @@
},
"outputs": [],
"source": [
"def substrings_in_string(big_string, substrings):\n",
"for substring in substrings:\n",
" if big_string.find(substring) != 1:\n",
" return substring\n",
"print(big_string)\n",
"return np.nan\n",
"#Turning cabin number into Deck\n",
"cabin_list = ['A', 'B', 'C', 'D', 'E', 'F', 'T', 'G', 'Unknown']\n",
"df['Deck']=df['Cabin'].map(lambda x: substrings_in_string(x, cabin_list))"

Loading…
Cancel
Save